Hugging Face's logo

saad7489
/
segformer-b0-finetuned-segments-SixrayKnife8-19-2024

Image Segmentation
TensorBoard
Safetensors
segformer
vision
Generated from Trainer
Model card Files Metrics Community
saad7489 commited on
Commit
6ab99ca
·
verified ·
1 Parent(s): 8c31662

End of training

Browse files
Files changed (5) hide show
  1. README.md +16 -16
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. runs/Aug20_09-40-25_d98ed18656c3/events.out.tfevents.1724146840.d98ed18656c3.1590.4 +3 -0
  5. training_args.bin +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
- base_model: nvidia/mit-b0
3
  license: other
 
4
  tags:
5
  - vision
6
  - image-segmentation
@@ -17,16 +17,16 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the saad7489/SixraygunTest dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.7621
21
- - Mean Iou: 0.4717
22
- - Mean Accuracy: 0.8197
23
- - Overall Accuracy: 0.8114
24
  - Accuracy Object1: nan
25
- - Accuracy Object2: 0.6867
26
- - Accuracy Object3: 0.9527
27
  - Iou Object1: 0.0
28
- - Iou Object2: 0.6649
29
- - Iou Object3: 0.7502
30
 
31
  ## Model description
32
 
@@ -57,13 +57,13 @@ The following hyperparameters were used during training:
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Object1 | Accuracy Object2 | Accuracy Object3 | Iou Object1 | Iou Object2 | Iou Object3 |
59
  |:-------------:|:------:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:----------------:|:----------------:|:----------------:|:-----------:|:-----------:|:-----------:|
60
- | 0.9327 | 1.4286 | 20 | 1.0058 | 0.3119 | 0.6386 | 0.6235 | nan | 0.3968 | 0.8803 | 0.0 | 0.3621 | 0.5736 |
61
- | 0.9028 | 2.8571 | 40 | 0.8719 | 0.3792 | 0.7295 | 0.7148 | nan | 0.4933 | 0.9656 | 0.0 | 0.4796 | 0.6580 |
62
- | 0.8514 | 4.2857 | 60 | 0.8211 | 0.4199 | 0.7734 | 0.7615 | nan | 0.5833 | 0.9635 | 0.0 | 0.5662 | 0.6936 |
63
- | 0.8251 | 5.7143 | 80 | 0.7680 | 0.4585 | 0.8077 | 0.7986 | nan | 0.6613 | 0.9541 | 0.0 | 0.6389 | 0.7364 |
64
- | 0.8048 | 7.1429 | 100 | 0.7591 | 0.4643 | 0.8141 | 0.8053 | nan | 0.6717 | 0.9565 | 0.0 | 0.6514 | 0.7414 |
65
- | 0.7667 | 8.5714 | 120 | 0.7591 | 0.4751 | 0.8230 | 0.8153 | nan | 0.6997 | 0.9462 | 0.0 | 0.6737 | 0.7517 |
66
- | 0.7748 | 10.0 | 140 | 0.7621 | 0.4717 | 0.8197 | 0.8114 | nan | 0.6867 | 0.9527 | 0.0 | 0.6649 | 0.7502 |
67
 
68
 
69
  ### Framework versions
 
1
  ---
 
2
  license: other
3
+ base_model: nvidia/mit-b0
4
  tags:
5
  - vision
6
  - image-segmentation
 
17
 
18
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the saad7489/SixraygunTest dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.5986
21
+ - Mean Iou: 0.5110
22
+ - Mean Accuracy: 0.8118
23
+ - Overall Accuracy: 0.8081
24
  - Accuracy Object1: nan
25
+ - Accuracy Object2: 0.7529
26
+ - Accuracy Object3: 0.8707
27
  - Iou Object1: 0.0
28
+ - Iou Object2: 0.7267
29
+ - Iou Object3: 0.8065
30
 
31
  ## Model description
32
 
 
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Object1 | Accuracy Object2 | Accuracy Object3 | Iou Object1 | Iou Object2 | Iou Object3 |
59
  |:-------------:|:------:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:----------------:|:----------------:|:----------------:|:-----------:|:-----------:|:-----------:|
60
+ | 0.7244 | 1.4286 | 20 | 0.6924 | 0.4978 | 0.8311 | 0.8263 | nan | 0.7536 | 0.9087 | 0.0 | 0.7153 | 0.7782 |
61
+ | 0.7251 | 2.8571 | 40 | 0.6470 | 0.5105 | 0.8256 | 0.8213 | nan | 0.7572 | 0.8940 | 0.0 | 0.7246 | 0.8069 |
62
+ | 0.6558 | 4.2857 | 60 | 0.6246 | 0.5148 | 0.8243 | 0.8210 | nan | 0.7714 | 0.8772 | 0.0 | 0.7350 | 0.8095 |
63
+ | 0.6463 | 5.7143 | 80 | 0.5954 | 0.5100 | 0.8126 | 0.8098 | nan | 0.7680 | 0.8572 | 0.0 | 0.7295 | 0.8004 |
64
+ | 0.63 | 7.1429 | 100 | 0.5892 | 0.5082 | 0.8073 | 0.8043 | nan | 0.7582 | 0.8564 | 0.0 | 0.7254 | 0.7991 |
65
+ | 0.5935 | 8.5714 | 120 | 0.5923 | 0.5134 | 0.8155 | 0.8127 | nan | 0.7694 | 0.8616 | 0.0 | 0.7369 | 0.8031 |
66
+ | 0.6137 | 10.0 | 140 | 0.5986 | 0.5110 | 0.8118 | 0.8081 | nan | 0.7529 | 0.8707 | 0.0 | 0.7267 | 0.8065 |
67
 
68
 
69
  ### Framework versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "nvidia/mit-b0",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
 
1
  {
2
+ "_name_or_path": "saad7489/segformer-b0-finetuned-segments-SixrayKnife8-19-2024",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:498d0290fa098046f6da81f26e0fcb0e5fd12a60754ef24708f9592d7a42d000
3
  size 14885804
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5dc1c0ab0356c5db2a4702b961ffcff17a830e114be7a268698492e0de171cd0
3
  size 14885804
runs/Aug20_09-40-25_d98ed18656c3/events.out.tfevents.1724146840.d98ed18656c3.1590.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34fff701c48822d1038515880c6f0dbc40fdfdd6e1806fbd01bbfb913f97ca4c
3
+ size 40320
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:388f19e064a52f447025ebda76dce089a17193998c81e56f334cce1462f11092
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6e1e7c3955d224ed1c9a7a0d31ce635711642d8bef84fdbfc5679e2d549991b
3
  size 5240