Hugging Face's logo

matei-dorian
/
segformer-b0-finetuned-human-parsing

Image Segmentation
Transformers
PyTorch
TensorBoard
Safetensors
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
2
matei-dorian commited on
Commit
a0a68f4
•
1 Parent(s): 4e0010c

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +101 -0
README.md ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ tags:
4
+ - vision
5
+ - image-segmentation
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: segformer-b0-finetuned-human-parsing
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # segformer-b0-finetuned-human-parsing
16
+
17
+ This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the None dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 1.9476
20
+ - Mean Iou: 0.0726
21
+ - Mean Accuracy: 0.1221
22
+ - Overall Accuracy: 0.3575
23
+ - Accuracy Background: nan
24
+ - Accuracy Hat: 0.0048
25
+ - Accuracy Hair: 0.4813
26
+ - Accuracy Sunglasses: 0.0
27
+ - Accuracy Upper-clothes: 0.9405
28
+ - Accuracy Skirt: 0.0000
29
+ - Accuracy Pants: 0.0631
30
+ - Accuracy Dress: 0.1031
31
+ - Accuracy Belt: 0.0
32
+ - Accuracy Left-shoe: 0.0011
33
+ - Accuracy Right-shoe: 0.0010
34
+ - Accuracy Face: 0.4406
35
+ - Accuracy Left-leg: 0.0291
36
+ - Accuracy Right-leg: 0.0
37
+ - Accuracy Left-arm: 0.0
38
+ - Accuracy Right-arm: 0.0001
39
+ - Accuracy Bag: 0.0114
40
+ - Accuracy Scarf: 0.0
41
+ - Iou Background: 0.0
42
+ - Iou Hat: 0.0043
43
+ - Iou Hair: 0.4221
44
+ - Iou Sunglasses: 0.0
45
+ - Iou Upper-clothes: 0.3239
46
+ - Iou Skirt: 0.0000
47
+ - Iou Pants: 0.0559
48
+ - Iou Dress: 0.0728
49
+ - Iou Belt: 0.0
50
+ - Iou Left-shoe: 0.0011
51
+ - Iou Right-shoe: 0.0009
52
+ - Iou Face: 0.3872
53
+ - Iou Left-leg: 0.0271
54
+ - Iou Right-leg: 0.0
55
+ - Iou Left-arm: 0.0
56
+ - Iou Right-arm: 0.0001
57
+ - Iou Bag: 0.0106
58
+ - Iou Scarf: 0.0
59
+
60
+ ## Model description
61
+
62
+ More information needed
63
+
64
+ ## Intended uses & limitations
65
+
66
+ More information needed
67
+
68
+ ## Training and evaluation data
69
+
70
+ More information needed
71
+
72
+ ## Training procedure
73
+
74
+ ### Training hyperparameters
75
+
76
+ The following hyperparameters were used during training:
77
+ - learning_rate: 6e-05
78
+ - train_batch_size: 2
79
+ - eval_batch_size: 2
80
+ - seed: 42
81
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
82
+ - lr_scheduler_type: linear
83
+ - num_epochs: 2
84
+
85
+ ### Training results
86
+
87
+ | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Hat | Accuracy Hair | Accuracy Sunglasses | Accuracy Upper-clothes | Accuracy Skirt | Accuracy Pants | Accuracy Dress | Accuracy Belt | Accuracy Left-shoe | Accuracy Right-shoe | Accuracy Face | Accuracy Left-leg | Accuracy Right-leg | Accuracy Left-arm | Accuracy Right-arm | Accuracy Bag | Accuracy Scarf | Iou Background | Iou Hat | Iou Hair | Iou Sunglasses | Iou Upper-clothes | Iou Skirt | Iou Pants | Iou Dress | Iou Belt | Iou Left-shoe | Iou Right-shoe | Iou Face | Iou Left-leg | Iou Right-leg | Iou Left-arm | Iou Right-arm | Iou Bag | Iou Scarf |
88
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:------------:|:-------------:|:-------------------:|:----------------------:|:--------------:|:--------------:|:--------------:|:-------------:|:------------------:|:-------------------:|:-------------:|:-----------------:|:------------------:|:-----------------:|:------------------:|:------------:|:--------------:|:--------------:|:-------:|:--------:|:--------------:|:-----------------:|:---------:|:---------:|:---------:|:--------:|:-------------:|:--------------:|:--------:|:------------:|:-------------:|:------------:|:-------------:|:-------:|:---------:|
89
+ | 2.5768 | 0.4 | 20 | 2.7812 | 0.0726 | 0.1332 | 0.2876 | nan | 0.0178 | 0.3204 | 0.0004 | 0.5548 | 0.0004 | 0.2555 | 0.2373 | 0.0 | 0.0103 | 0.0003 | 0.5637 | 0.0287 | 0.0302 | 0.0001 | 0.0008 | 0.2435 | 0.0 | 0.0 | 0.0166 | 0.2759 | 0.0001 | 0.2781 | 0.0004 | 0.1710 | 0.1295 | 0.0 | 0.0098 | 0.0003 | 0.3251 | 0.0260 | 0.0248 | 0.0001 | 0.0007 | 0.0491 | 0.0 |
90
+ | 2.2093 | 0.8 | 40 | 2.5166 | 0.0563 | 0.1052 | 0.3288 | nan | 0.0 | 0.1994 | 0.0 | 0.9447 | 0.0015 | 0.0435 | 0.1164 | 0.0 | 0.0008 | 0.0000 | 0.4655 | 0.0007 | 0.0003 | 0.0 | 0.0 | 0.0153 | 0.0 | 0.0 | 0.0 | 0.1946 | 0.0 | 0.3037 | 0.0015 | 0.0417 | 0.0842 | 0.0 | 0.0008 | 0.0000 | 0.3726 | 0.0007 | 0.0003 | 0.0 | 0.0 | 0.0124 | 0.0 |
91
+ | 1.8804 | 1.2 | 60 | 2.0209 | 0.0632 | 0.1110 | 0.3374 | nan | 0.0087 | 0.3724 | 0.0 | 0.9475 | 0.0014 | 0.0162 | 0.0528 | 0.0 | 0.0001 | 0.0008 | 0.4257 | 0.0561 | 0.0001 | 0.0 | 0.0 | 0.0055 | 0.0 | 0.0 | 0.0077 | 0.3472 | 0.0 | 0.3086 | 0.0014 | 0.0156 | 0.0403 | 0.0 | 0.0001 | 0.0008 | 0.3597 | 0.0515 | 0.0001 | 0.0 | 0.0 | 0.0052 | 0.0 |
92
+ | 1.8776 | 1.6 | 80 | 2.0016 | 0.0665 | 0.1154 | 0.3454 | nan | 0.0056 | 0.4172 | 0.0 | 0.9412 | 0.0000 | 0.0490 | 0.0697 | 0.0 | 0.0002 | 0.0006 | 0.4349 | 0.0329 | 0.0000 | 0.0 | 0.0000 | 0.0100 | 0.0 | 0.0 | 0.0048 | 0.3791 | 0.0 | 0.3138 | 0.0000 | 0.0438 | 0.0542 | 0.0 | 0.0002 | 0.0006 | 0.3608 | 0.0304 | 0.0000 | 0.0 | 0.0000 | 0.0093 | 0.0 |
93
+ | 1.8471 | 2.0 | 100 | 1.9476 | 0.0726 | 0.1221 | 0.3575 | nan | 0.0048 | 0.4813 | 0.0 | 0.9405 | 0.0000 | 0.0631 | 0.1031 | 0.0 | 0.0011 | 0.0010 | 0.4406 | 0.0291 | 0.0 | 0.0 | 0.0001 | 0.0114 | 0.0 | 0.0 | 0.0043 | 0.4221 | 0.0 | 0.3239 | 0.0000 | 0.0559 | 0.0728 | 0.0 | 0.0011 | 0.0009 | 0.3872 | 0.0271 | 0.0 | 0.0 | 0.0001 | 0.0106 | 0.0 |
94
+
95
+
96
+ ### Framework versions
97
+
98
+ - Transformers 4.28.1
99
+ - Pytorch 2.0.0+cu118
100
+ - Datasets 2.12.0
101
+ - Tokenizers 0.13.3