Hugging Face's logo

topguns
/
segformer-b0-finetuned-segments-sidewalk-outputs

Image Segmentation
Transformers
TensorBoard
Safetensors
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
topguns commited on
Commit
4c0a4f2
•
1 Parent(s): 008fe06

End of training

Browse files
Files changed (7) hide show
  1. README.md +102 -0
  2. config.json +80 -0
  3. model.safetensors +3 -0
  4. runs/Dec06_20-59-28_36ed952463d5/events.out.tfevents.1733518821.36ed952463d5.208.0 +3 -0
  5. runs/Dec06_21-13-58_36ed952463d5/events.out.tfevents.1733519668.36ed952463d5.208.1 +3 -0
  6. runs/Dec06_22-03-36_36ed952463d5/events.out.tfevents.1733522620.36ed952463d5.208.2 +3 -0
  7. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: other
4
+ base_model: nvidia/mit-b0
5
+ tags:
6
+ - vision
7
+ - image-segmentation
8
+ - generated_from_trainer
9
+ model-index:
10
+ - name: segformer-b0-finetuned-segments-sidewalk-outputs
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # segformer-b0-finetuned-segments-sidewalk-outputs
18
+
19
+ This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the segments/sidewalk-semantic dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.0650
22
+ - Mean Iou: 0.5841
23
+ - Mean Accuracy: 0.8778
24
+ - Overall Accuracy: 0.9588
25
+ - Accuracy Background: nan
26
+ - Accuracy Ground: 0.9811
27
+ - Accuracy Pallet: 0.7744
28
+ - Iou Background: 0.0
29
+ - Iou Ground: 0.9805
30
+ - Iou Pallet: 0.7717
31
+
32
+ ## Model description
33
+
34
+ More information needed
35
+
36
+ ## Intended uses & limitations
37
+
38
+ More information needed
39
+
40
+ ## Training and evaluation data
41
+
42
+ More information needed
43
+
44
+ ## Training procedure
45
+
46
+ ### Training hyperparameters
47
+
48
+ The following hyperparameters were used during training:
49
+ - learning_rate: 6e-05
50
+ - train_batch_size: 2
51
+ - eval_batch_size: 2
52
+ - seed: 42
53
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
54
+ - lr_scheduler_type: linear
55
+ - num_epochs: 5
56
+
57
+ ### Training results
58
+
59
+ | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Ground | Accuracy Pallet | Iou Background | Iou Ground | Iou Pallet |
60
+ |:-------------:|:------:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:---------------:|:---------------:|:--------------:|:----------:|:----------:|
61
+ | 0.0421 | 0.1471 | 20 | 0.0709 | 0.5844 | 0.8798 | 0.9604 | nan | 0.9826 | 0.7771 | 0.0 | 0.9808 | 0.7724 |
62
+ | 0.0276 | 0.2941 | 40 | 0.0624 | 0.6015 | 0.9042 | 0.9646 | nan | 0.9813 | 0.8271 | 0.0 | 0.9809 | 0.8236 |
63
+ | 0.0211 | 0.4412 | 60 | 0.0691 | 0.5664 | 0.8509 | 0.9552 | nan | 0.9840 | 0.7179 | 0.0 | 0.9835 | 0.7157 |
64
+ | 0.0235 | 0.5882 | 80 | 0.0621 | 0.5617 | 0.8437 | 0.9565 | nan | 0.9876 | 0.6998 | 0.0 | 0.9866 | 0.6984 |
65
+ | 0.0271 | 0.7353 | 100 | 0.0602 | 0.5877 | 0.8832 | 0.9567 | nan | 0.9770 | 0.7893 | 0.0 | 0.9767 | 0.7863 |
66
+ | 0.024 | 0.8824 | 120 | 0.0682 | 0.5923 | 0.8911 | 0.9600 | nan | 0.9790 | 0.8033 | 0.0 | 0.9785 | 0.7985 |
67
+ | 0.0407 | 1.0294 | 140 | 0.0691 | 0.5892 | 0.8859 | 0.9580 | nan | 0.9779 | 0.7940 | 0.0 | 0.9773 | 0.7903 |
68
+ | 0.0277 | 1.1765 | 160 | 0.0683 | 0.5788 | 0.8697 | 0.9571 | nan | 0.9812 | 0.7582 | 0.0 | 0.9804 | 0.7560 |
69
+ | 0.017 | 1.3235 | 180 | 0.0679 | 0.5845 | 0.8789 | 0.9566 | nan | 0.9780 | 0.7798 | 0.0 | 0.9770 | 0.7764 |
70
+ | 0.0275 | 1.4706 | 200 | 0.0634 | 0.5992 | 0.9014 | 0.9639 | nan | 0.9812 | 0.8216 | 0.0 | 0.9800 | 0.8175 |
71
+ | 0.0122 | 1.6176 | 220 | 0.0602 | 0.5859 | 0.8813 | 0.9590 | nan | 0.9804 | 0.7821 | 0.0 | 0.9794 | 0.7782 |
72
+ | 0.0149 | 1.7647 | 240 | 0.0662 | 0.5827 | 0.8757 | 0.9591 | nan | 0.9820 | 0.7694 | 0.0 | 0.9817 | 0.7664 |
73
+ | 0.0169 | 1.9118 | 260 | 0.0628 | 0.5994 | 0.9019 | 0.9614 | nan | 0.9778 | 0.8259 | 0.0 | 0.9776 | 0.8205 |
74
+ | 0.0324 | 2.0588 | 280 | 0.0677 | 0.5859 | 0.8809 | 0.9584 | nan | 0.9798 | 0.7819 | 0.0 | 0.9792 | 0.7785 |
75
+ | 0.0229 | 2.2059 | 300 | 0.0693 | 0.5983 | 0.9003 | 0.9619 | nan | 0.9789 | 0.8217 | 0.0 | 0.9784 | 0.8166 |
76
+ | 0.0204 | 2.3529 | 320 | 0.0729 | 0.5850 | 0.8792 | 0.9586 | nan | 0.9805 | 0.7780 | 0.0 | 0.9800 | 0.7749 |
77
+ | 0.0102 | 2.5 | 340 | 0.0655 | 0.5899 | 0.8868 | 0.9603 | nan | 0.9806 | 0.7931 | 0.0 | 0.9802 | 0.7895 |
78
+ | 0.0235 | 2.6471 | 360 | 0.0682 | 0.5781 | 0.8688 | 0.9556 | nan | 0.9795 | 0.7580 | 0.0 | 0.9789 | 0.7552 |
79
+ | 0.0239 | 2.7941 | 380 | 0.0633 | 0.5961 | 0.8965 | 0.9615 | nan | 0.9794 | 0.8136 | 0.0 | 0.9789 | 0.8093 |
80
+ | 0.0305 | 2.9412 | 400 | 0.0593 | 0.5832 | 0.8764 | 0.9593 | nan | 0.9822 | 0.7706 | 0.0 | 0.9817 | 0.7678 |
81
+ | 0.0183 | 3.0882 | 420 | 0.0600 | 0.5867 | 0.8816 | 0.9613 | nan | 0.9832 | 0.7799 | 0.0 | 0.9827 | 0.7775 |
82
+ | 0.031 | 3.2353 | 440 | 0.0612 | 0.5933 | 0.8917 | 0.9614 | nan | 0.9806 | 0.8029 | 0.0 | 0.9799 | 0.8000 |
83
+ | 0.0174 | 3.3824 | 460 | 0.0645 | 0.5836 | 0.8769 | 0.9590 | nan | 0.9816 | 0.7722 | 0.0 | 0.9811 | 0.7696 |
84
+ | 0.0456 | 3.5294 | 480 | 0.0651 | 0.5770 | 0.8669 | 0.9577 | nan | 0.9827 | 0.7512 | 0.0 | 0.9821 | 0.7489 |
85
+ | 0.0187 | 3.6765 | 500 | 0.0659 | 0.5831 | 0.8765 | 0.9578 | nan | 0.9803 | 0.7727 | 0.0 | 0.9798 | 0.7695 |
86
+ | 0.0329 | 3.8235 | 520 | 0.0690 | 0.5787 | 0.8697 | 0.9568 | nan | 0.9808 | 0.7587 | 0.0 | 0.9801 | 0.7560 |
87
+ | 0.0241 | 3.9706 | 540 | 0.0651 | 0.5847 | 0.8789 | 0.9584 | nan | 0.9803 | 0.7774 | 0.0 | 0.9798 | 0.7743 |
88
+ | 0.0304 | 4.1176 | 560 | 0.0652 | 0.5871 | 0.8823 | 0.9589 | nan | 0.9800 | 0.7846 | 0.0 | 0.9795 | 0.7817 |
89
+ | 0.0086 | 4.2647 | 580 | 0.0662 | 0.5851 | 0.8793 | 0.9584 | nan | 0.9802 | 0.7784 | 0.0 | 0.9797 | 0.7756 |
90
+ | 0.0194 | 4.4118 | 600 | 0.0678 | 0.5889 | 0.8853 | 0.9581 | nan | 0.9781 | 0.7925 | 0.0 | 0.9777 | 0.7890 |
91
+ | 0.0114 | 4.5588 | 620 | 0.0664 | 0.5877 | 0.8834 | 0.9582 | nan | 0.9789 | 0.7880 | 0.0 | 0.9784 | 0.7847 |
92
+ | 0.0183 | 4.7059 | 640 | 0.0663 | 0.5843 | 0.8782 | 0.9571 | nan | 0.9789 | 0.7776 | 0.0 | 0.9784 | 0.7744 |
93
+ | 0.0139 | 4.8529 | 660 | 0.0652 | 0.5872 | 0.8826 | 0.9590 | nan | 0.9801 | 0.7852 | 0.0 | 0.9796 | 0.7820 |
94
+ | 0.0197 | 5.0 | 680 | 0.0650 | 0.5841 | 0.8778 | 0.9588 | nan | 0.9811 | 0.7744 | 0.0 | 0.9805 | 0.7717 |
95
+
96
+
97
+ ### Framework versions
98
+
99
+ - Transformers 4.46.3
100
+ - Pytorch 2.5.1+cu121
101
+ - Datasets 3.1.0
102
+ - Tokenizers 0.20.3
config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b0",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 256,
9
+ "depths": [
10
+ 2,
11
+ 2,
12
+ 2,
13
+ 2
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 32,
26
+ 64,
27
+ 160,
28
+ 256
29
+ ],
30
+ "id2label": {
31
+ "0": "background",
32
+ "1": "ground",
33
+ "2": "pallet"
34
+ },
35
+ "image_size": 224,
36
+ "initializer_range": 0.02,
37
+ "label2id": {
38
+ "background": 0,
39
+ "ground": 1,
40
+ "pallet": 2
41
+ },
42
+ "layer_norm_eps": 1e-06,
43
+ "mlp_ratios": [
44
+ 4,
45
+ 4,
46
+ 4,
47
+ 4
48
+ ],
49
+ "model_type": "segformer",
50
+ "num_attention_heads": [
51
+ 1,
52
+ 2,
53
+ 5,
54
+ 8
55
+ ],
56
+ "num_channels": 3,
57
+ "num_encoder_blocks": 4,
58
+ "patch_sizes": [
59
+ 7,
60
+ 3,
61
+ 3,
62
+ 3
63
+ ],
64
+ "reshape_last_stage": true,
65
+ "semantic_loss_ignore_index": 255,
66
+ "sr_ratios": [
67
+ 8,
68
+ 4,
69
+ 2,
70
+ 1
71
+ ],
72
+ "strides": [
73
+ 4,
74
+ 2,
75
+ 2,
76
+ 2
77
+ ],
78
+ "torch_dtype": "float32",
79
+ "transformers_version": "4.46.3"
80
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9dd8e8ab7ee867c822a20618558b89def3f2543330b2724bc91ac1a5bb1d50c6
3
+ size 14885804
runs/Dec06_20-59-28_36ed952463d5/events.out.tfevents.1733518821.36ed952463d5.208.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2e1bfc352a26abf708d261ef261bd3542a056a1f90838aa9595f2cf03349bbf
3
+ size 5638
runs/Dec06_21-13-58_36ed952463d5/events.out.tfevents.1733519668.36ed952463d5.208.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7cf839cec33618e409b9701db822019b4ac9898875f9faea4509662d7f0b71d
3
+ size 1358835
runs/Dec06_22-03-36_36ed952463d5/events.out.tfevents.1733522620.36ed952463d5.208.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e953de9ec6997757f0f226e8b719648772926fa05a9795b2a54f27f2aaadd58
3
+ size 175598
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b2a5fc5fef9dd45329568712d65a53b12c91b34f9c4013f71c75e21c1d1580a
3
+ size 5304