Hugging Face's logo

peldrak
/
segformer-b4-cityscapes-finetuned-coastTrain

Image Segmentation
Transformers
TensorBoard
Safetensors
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
peldrak commited on
Commit
f6d8282
•
1 Parent(s): 8e426e7

End of training

Browse files
Files changed (4) hide show
  1. README.md +62 -88
  2. model.safetensors +1 -1
  3. runs/Mar04_09-24-15_1fcd91d1db08/events.out.tfevents.1709544258.1fcd91d1db08.27.0 +3 -0
  4. training_args.bin +1 -1
README.md CHANGED
@@ -1,10 +1,10 @@
1
  ---
2
  license: other
 
3
  tags:
4
  - vision
5
  - image-segmentation
6
  - generated_from_trainer
7
- base_model: nvidia/segformer-b4-finetuned-cityscapes-1024-1024
8
  model-index:
9
  - name: segformer-b4-cityscapes-finetuned-coastTrain
10
  results: []
@@ -17,26 +17,25 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [nvidia/segformer-b4-finetuned-cityscapes-1024-1024](https://huggingface.co/nvidia/segformer-b4-finetuned-cityscapes-1024-1024) on the peldrak/coastTrain dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.2425
21
- - Mean Iou: 0.7753
22
- - Mean Accuracy: 0.8403
23
- - Overall Accuracy: 0.9317
24
- - Accuracy Water: 0.9712
25
- - Accuracy Whitewater: 0.5496
26
- - Accuracy Sediment: 0.9007
27
- - Accuracy Other Natural Terrain: 0.6504
28
- - Accuracy Vegetation: 0.9190
29
- - Accuracy Development: 0.9343
30
- - Accuracy Unknown: 0.9569
31
- - Iou Water: 0.9249
32
- - Iou Whitewater: 0.4929
33
- - Iou Sediment: 0.8001
34
- - Iou Other Natural Terrain: 0.5731
35
- - Iou Vegetation: 0.8521
36
- - Iou Development: 0.8354
37
- - Iou Unknown: 0.9485
38
- - F1 Score: 0.9308
39
- - Dice Coefficient: 4.5267
40
 
41
  ## Model description
42
 
@@ -61,76 +60,51 @@ The following hyperparameters were used during training:
61
  - seed: 42
62
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
63
  - lr_scheduler_type: linear
64
- - num_epochs: 10
65
 
66
  ### Training results
67
 
68
- | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Water | Accuracy Whitewater | Accuracy Sediment | Accuracy Other Natural Terrain | Accuracy Vegetation | Accuracy Development | Accuracy Unknown | Iou Water | Iou Whitewater | Iou Sediment | Iou Other Natural Terrain | Iou Vegetation | Iou Development | Iou Unknown | F1 Score | Dice Coefficient |
69
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:--------------:|:-------------------:|:-----------------:|:------------------------------:|:-------------------:|:--------------------:|:----------------:|:---------:|:--------------:|:------------:|:-------------------------:|:--------------:|:---------------:|:-----------:|:--------:|:----------------:|
70
- | 1.5962 | 0.16 | 20 | 1.4428 | 0.3376 | 0.4422 | 0.6647 | 0.7798 | 0.0000 | 0.3962 | 0.0066 | 0.8702 | 0.4492 | 0.5931 | 0.5924 | 0.0000 | 0.3276 | 0.0063 | 0.5046 | 0.3409 | 0.5916 | 0.6498 | 3.7438 |
71
- | 1.3982 | 0.31 | 40 | 1.1292 | 0.3903 | 0.5003 | 0.7122 | 0.7617 | 0.0 | 0.4647 | 0.0 | 0.9527 | 0.6896 | 0.6333 | 0.6491 | 0.0 | 0.4124 | 0.0 | 0.5404 | 0.4972 | 0.6328 | 0.7015 | 3.9010 |
72
- | 1.076 | 0.47 | 60 | 0.8926 | 0.4810 | 0.5758 | 0.8063 | 0.8345 | 0.0 | 0.6712 | 0.0 | 0.9253 | 0.7242 | 0.8757 | 0.7734 | 0.0 | 0.5271 | 0.0 | 0.6512 | 0.5461 | 0.8693 | 0.7971 | 4.3369 |
73
- | 0.9392 | 0.62 | 80 | 0.7618 | 0.4856 | 0.5812 | 0.8070 | 0.8410 | 0.0 | 0.7868 | 0.0 | 0.8496 | 0.7203 | 0.8706 | 0.7542 | 0.0 | 0.5382 | 0.0 | 0.6715 | 0.5687 | 0.8668 | 0.7990 | 4.3038 |
74
- | 1.0916 | 0.78 | 100 | 0.6937 | 0.4627 | 0.5615 | 0.7893 | 0.9559 | 0.0 | 0.4611 | 0.0 | 0.7796 | 0.8447 | 0.8894 | 0.7178 | 0.0 | 0.3996 | 0.0 | 0.6583 | 0.5825 | 0.8812 | 0.7707 | 4.3804 |
75
- | 0.6819 | 0.93 | 120 | 0.5478 | 0.5172 | 0.6107 | 0.8423 | 0.9306 | 0.0 | 0.7898 | 0.0 | 0.8249 | 0.8434 | 0.8864 | 0.8348 | 0.0 | 0.6075 | 0.0 | 0.6964 | 0.6030 | 0.8790 | 0.8317 | 4.4022 |
76
- | 0.7826 | 1.09 | 140 | 0.5417 | 0.5308 | 0.6106 | 0.8535 | 0.9110 | 0.0 | 0.7538 | 0.0 | 0.9260 | 0.7791 | 0.9044 | 0.8655 | 0.0 | 0.6341 | 0.0 | 0.6924 | 0.6339 | 0.8897 | 0.8428 | 4.4211 |
77
- | 1.0433 | 1.24 | 160 | 0.5368 | 0.5317 | 0.6136 | 0.8539 | 0.9249 | 0.0 | 0.7169 | 0.0 | 0.9250 | 0.8354 | 0.8931 | 0.8589 | 0.0 | 0.6248 | 0.0 | 0.7009 | 0.6503 | 0.8867 | 0.8424 | 4.4259 |
78
- | 0.9705 | 1.4 | 180 | 0.4719 | 0.5298 | 0.6285 | 0.8510 | 0.9036 | 0.0000 | 0.8172 | 0.0 | 0.8476 | 0.9337 | 0.8974 | 0.8354 | 0.0000 | 0.6234 | 0.0 | 0.7268 | 0.6336 | 0.8893 | 0.8413 | 4.4397 |
79
- | 0.3833 | 1.55 | 200 | 0.4593 | 0.5309 | 0.6301 | 0.8550 | 0.8881 | 0.0 | 0.7657 | 0.0 | 0.9061 | 0.9373 | 0.9136 | 0.8379 | 0.0 | 0.6371 | 0.0 | 0.7425 | 0.5985 | 0.9000 | 0.8452 | 4.4467 |
80
- | 0.8535 | 1.71 | 220 | 0.4276 | 0.5407 | 0.6311 | 0.8617 | 0.9337 | 0.0000 | 0.7790 | 0.0 | 0.8663 | 0.9258 | 0.9126 | 0.8703 | 0.0000 | 0.6404 | 0.0 | 0.7161 | 0.6591 | 0.8993 | 0.8505 | 4.4691 |
81
- | 0.3945 | 1.86 | 240 | 0.4641 | 0.5296 | 0.6253 | 0.8466 | 0.8523 | 0.0003 | 0.7567 | 0.0 | 0.9279 | 0.9228 | 0.9173 | 0.8100 | 0.0003 | 0.6136 | 0.0 | 0.7175 | 0.6599 | 0.9056 | 0.8364 | 4.4176 |
82
- | 0.2928 | 2.02 | 260 | 0.3834 | 0.5654 | 0.6377 | 0.8787 | 0.9302 | 0.0003 | 0.7733 | 0.0 | 0.9370 | 0.8837 | 0.9397 | 0.8650 | 0.0003 | 0.6523 | 0.0 | 0.7696 | 0.7418 | 0.9290 | 0.8661 | 4.4851 |
83
- | 0.6195 | 2.17 | 280 | 0.3680 | 0.5545 | 0.6373 | 0.8746 | 0.9397 | 0.0002 | 0.7829 | 0.0 | 0.9151 | 0.9151 | 0.9081 | 0.8711 | 0.0002 | 0.6773 | 0.0 | 0.7602 | 0.6727 | 0.9000 | 0.8627 | 4.4820 |
84
- | 0.4975 | 2.33 | 300 | 0.3619 | 0.5697 | 0.6458 | 0.8842 | 0.9467 | 0.0007 | 0.8135 | 0.0 | 0.9180 | 0.9287 | 0.9132 | 0.8824 | 0.0007 | 0.6910 | 0.0 | 0.7755 | 0.7357 | 0.9030 | 0.8719 | 4.4852 |
85
- | 0.4836 | 2.48 | 320 | 0.3661 | 0.5598 | 0.6398 | 0.8796 | 0.9505 | 0.0030 | 0.8075 | 0.0 | 0.9060 | 0.8994 | 0.9124 | 0.8894 | 0.0030 | 0.6882 | 0.0 | 0.7595 | 0.6750 | 0.9034 | 0.8677 | 4.4812 |
86
- | 1.1391 | 2.64 | 340 | 0.3702 | 0.5681 | 0.6483 | 0.8843 | 0.9572 | 0.0125 | 0.8592 | 0.0001 | 0.8737 | 0.9182 | 0.9172 | 0.8830 | 0.0125 | 0.7157 | 0.0001 | 0.7720 | 0.6888 | 0.9045 | 0.8725 | 4.4865 |
87
- | 1.3634 | 2.79 | 360 | 0.3349 | 0.5827 | 0.6550 | 0.8952 | 0.9513 | 0.0018 | 0.8981 | 0.0 | 0.9011 | 0.9111 | 0.9216 | 0.8927 | 0.0018 | 0.7293 | 0.0 | 0.8045 | 0.7444 | 0.9063 | 0.8830 | 4.4794 |
88
- | 0.6232 | 2.95 | 380 | 0.4271 | 0.5737 | 0.6576 | 0.8786 | 0.8963 | 0.0178 | 0.9155 | 0.0264 | 0.8670 | 0.9264 | 0.9538 | 0.8596 | 0.0178 | 0.6854 | 0.0264 | 0.7630 | 0.7343 | 0.9297 | 0.8687 | 4.4571 |
89
- | 0.3261 | 3.1 | 400 | 0.3845 | 0.5850 | 0.6563 | 0.8922 | 0.9487 | 0.0189 | 0.9155 | 0.0001 | 0.8591 | 0.8953 | 0.9567 | 0.8951 | 0.0189 | 0.7002 | 0.0001 | 0.7758 | 0.7601 | 0.9451 | 0.8808 | 4.4859 |
90
- | 0.377 | 3.26 | 420 | 0.3382 | 0.5933 | 0.6623 | 0.8936 | 0.9698 | 0.0305 | 0.8218 | 0.0571 | 0.8922 | 0.9156 | 0.9490 | 0.8880 | 0.0305 | 0.7024 | 0.0570 | 0.7852 | 0.7480 | 0.9418 | 0.8822 | 4.5070 |
91
- | 0.5478 | 3.41 | 440 | 0.3242 | 0.5987 | 0.6638 | 0.9005 | 0.9451 | 0.0199 | 0.8625 | 0.0413 | 0.9362 | 0.8918 | 0.9500 | 0.8944 | 0.0199 | 0.7281 | 0.0413 | 0.8074 | 0.7591 | 0.9412 | 0.8891 | 4.5042 |
92
- | 0.7848 | 3.57 | 460 | 0.3639 | 0.5934 | 0.6611 | 0.8915 | 0.9506 | 0.0815 | 0.9009 | 0.0 | 0.8865 | 0.8854 | 0.9226 | 0.8762 | 0.0810 | 0.7116 | 0.0 | 0.8005 | 0.7825 | 0.9022 | 0.8809 | 4.4382 |
93
- | 0.5094 | 3.72 | 480 | 0.3168 | 0.6260 | 0.6939 | 0.9020 | 0.9609 | 0.0960 | 0.8004 | 0.1851 | 0.9374 | 0.9270 | 0.9502 | 0.8971 | 0.0955 | 0.7399 | 0.1829 | 0.8029 | 0.7263 | 0.9371 | 0.8937 | 4.5159 |
94
- | 0.2645 | 3.88 | 500 | 0.3101 | 0.6201 | 0.6938 | 0.9037 | 0.9633 | 0.0827 | 0.8948 | 0.1505 | 0.8996 | 0.9448 | 0.9206 | 0.9063 | 0.0822 | 0.7623 | 0.1497 | 0.8170 | 0.7113 | 0.9117 | 0.8956 | 4.5174 |
95
- | 0.2297 | 4.03 | 520 | 0.3096 | 0.6455 | 0.7142 | 0.9029 | 0.9296 | 0.1769 | 0.8964 | 0.2131 | 0.9366 | 0.9253 | 0.9216 | 0.8869 | 0.1732 | 0.7359 | 0.2104 | 0.8289 | 0.7712 | 0.9119 | 0.8972 | 4.4938 |
96
- | 0.391 | 4.19 | 540 | 0.3010 | 0.6211 | 0.6824 | 0.9070 | 0.9695 | 0.1072 | 0.8558 | 0.0643 | 0.9273 | 0.9109 | 0.9420 | 0.8983 | 0.1063 | 0.7297 | 0.0643 | 0.8350 | 0.7815 | 0.9325 | 0.8972 | 4.5162 |
97
- | 0.499 | 4.34 | 560 | 0.3321 | 0.6496 | 0.7187 | 0.9027 | 0.9204 | 0.2130 | 0.9018 | 0.1887 | 0.9250 | 0.9381 | 0.9435 | 0.8801 | 0.2077 | 0.7137 | 0.1860 | 0.8345 | 0.7913 | 0.9341 | 0.8976 | 4.4911 |
98
- | 0.35 | 4.5 | 580 | 0.2716 | 0.6865 | 0.7407 | 0.9169 | 0.9744 | 0.2361 | 0.8576 | 0.3179 | 0.9205 | 0.9139 | 0.9647 | 0.9034 | 0.2285 | 0.7564 | 0.3053 | 0.8324 | 0.8283 | 0.9513 | 0.9117 | 4.5262 |
99
- | 0.4687 | 4.65 | 600 | 0.2896 | 0.6770 | 0.7403 | 0.9107 | 0.9715 | 0.2743 | 0.8350 | 0.3159 | 0.9419 | 0.9263 | 0.9169 | 0.9077 | 0.2605 | 0.7528 | 0.3051 | 0.8288 | 0.7750 | 0.9088 | 0.9061 | 4.5175 |
100
- | 0.2071 | 4.81 | 620 | 0.2655 | 0.7097 | 0.7685 | 0.9218 | 0.9571 | 0.2874 | 0.8711 | 0.4246 | 0.9398 | 0.9337 | 0.9661 | 0.9146 | 0.2755 | 0.7707 | 0.3984 | 0.8409 | 0.8184 | 0.9496 | 0.9182 | 4.5296 |
101
- | 0.1526 | 4.96 | 640 | 0.2755 | 0.6872 | 0.7412 | 0.9183 | 0.9677 | 0.1657 | 0.9126 | 0.3708 | 0.9079 | 0.9055 | 0.9586 | 0.9044 | 0.1629 | 0.7685 | 0.3607 | 0.8364 | 0.8284 | 0.9491 | 0.9129 | 4.5111 |
102
- | 0.2048 | 5.12 | 660 | 0.2713 | 0.6881 | 0.7479 | 0.9160 | 0.9777 | 0.1911 | 0.8571 | 0.3990 | 0.9088 | 0.9467 | 0.9547 | 0.8974 | 0.1877 | 0.7635 | 0.3842 | 0.8379 | 0.7996 | 0.9465 | 0.9109 | 4.5235 |
103
- | 0.3091 | 5.27 | 680 | 0.2704 | 0.7047 | 0.7635 | 0.9199 | 0.9547 | 0.3000 | 0.8839 | 0.3843 | 0.9394 | 0.9292 | 0.9531 | 0.9066 | 0.2859 | 0.7795 | 0.3738 | 0.8411 | 0.8005 | 0.9454 | 0.9164 | 4.5114 |
104
- | 0.261 | 5.43 | 700 | 0.2615 | 0.7082 | 0.7657 | 0.9220 | 0.9623 | 0.2893 | 0.8987 | 0.3998 | 0.9274 | 0.9286 | 0.9537 | 0.9126 | 0.2783 | 0.7720 | 0.3825 | 0.8485 | 0.8181 | 0.9452 | 0.9184 | 4.5175 |
105
- | 0.3682 | 5.58 | 720 | 0.2590 | 0.7245 | 0.7870 | 0.9224 | 0.9701 | 0.3070 | 0.8708 | 0.5413 | 0.9263 | 0.9443 | 0.9487 | 0.9106 | 0.2935 | 0.7881 | 0.5005 | 0.8439 | 0.7925 | 0.9423 | 0.9196 | 4.5242 |
106
- | 0.156 | 5.74 | 740 | 0.2621 | 0.7348 | 0.8011 | 0.9222 | 0.9658 | 0.3967 | 0.9061 | 0.5362 | 0.8881 | 0.9460 | 0.9691 | 0.9189 | 0.3659 | 0.7698 | 0.5022 | 0.8324 | 0.7987 | 0.9554 | 0.9204 | 4.5275 |
107
- | 0.1901 | 5.89 | 760 | 0.2475 | 0.7312 | 0.7870 | 0.9262 | 0.9661 | 0.3411 | 0.8744 | 0.5077 | 0.9411 | 0.9127 | 0.9658 | 0.9183 | 0.3218 | 0.7721 | 0.4707 | 0.8531 | 0.8278 | 0.9544 | 0.9234 | 4.5318 |
108
- | 1.0784 | 6.05 | 780 | 0.2827 | 0.7124 | 0.7731 | 0.9209 | 0.9657 | 0.3051 | 0.9162 | 0.4400 | 0.9069 | 0.9290 | 0.9488 | 0.9163 | 0.2907 | 0.7708 | 0.4173 | 0.8401 | 0.8084 | 0.9432 | 0.9179 | 4.5110 |
109
- | 0.2541 | 6.2 | 800 | 0.2548 | 0.7340 | 0.7956 | 0.9248 | 0.9699 | 0.3754 | 0.8811 | 0.5368 | 0.9280 | 0.9263 | 0.9520 | 0.9150 | 0.3427 | 0.7824 | 0.4843 | 0.8483 | 0.8200 | 0.9452 | 0.9225 | 4.5209 |
110
- | 0.3264 | 6.36 | 820 | 0.2459 | 0.7410 | 0.8080 | 0.9256 | 0.9698 | 0.4233 | 0.8688 | 0.5649 | 0.9239 | 0.9449 | 0.9601 | 0.9208 | 0.3799 | 0.7794 | 0.5040 | 0.8464 | 0.8058 | 0.9505 | 0.9238 | 4.5347 |
111
- | 0.3072 | 6.51 | 840 | 0.2415 | 0.7482 | 0.8198 | 0.9267 | 0.9676 | 0.3956 | 0.9116 | 0.6646 | 0.9055 | 0.9390 | 0.9550 | 0.9226 | 0.3653 | 0.7900 | 0.5598 | 0.8477 | 0.8052 | 0.9466 | 0.9253 | 4.5245 |
112
- | 0.2599 | 6.67 | 860 | 0.2562 | 0.7483 | 0.8108 | 0.9273 | 0.9699 | 0.4422 | 0.8566 | 0.5772 | 0.9349 | 0.9236 | 0.9710 | 0.9225 | 0.3994 | 0.7795 | 0.5122 | 0.8439 | 0.8237 | 0.9568 | 0.9254 | 4.5406 |
113
- | 0.1983 | 6.82 | 880 | 0.2446 | 0.7477 | 0.8129 | 0.9267 | 0.9665 | 0.4592 | 0.8919 | 0.5685 | 0.9229 | 0.9225 | 0.9590 | 0.9231 | 0.4085 | 0.7810 | 0.5059 | 0.8470 | 0.8195 | 0.9491 | 0.9253 | 4.5257 |
114
- | 0.3021 | 6.98 | 900 | 0.2584 | 0.7256 | 0.7832 | 0.9233 | 0.9716 | 0.3915 | 0.8599 | 0.4415 | 0.9392 | 0.9292 | 0.9491 | 0.9135 | 0.3623 | 0.7707 | 0.4166 | 0.8488 | 0.8242 | 0.9427 | 0.9205 | 4.5221 |
115
- | 0.2292 | 7.13 | 920 | 0.2488 | 0.7355 | 0.7939 | 0.9269 | 0.9657 | 0.3814 | 0.8895 | 0.4974 | 0.9373 | 0.9344 | 0.9518 | 0.9207 | 0.3580 | 0.7844 | 0.4590 | 0.8523 | 0.8288 | 0.9456 | 0.9246 | 4.5211 |
116
- | 0.2052 | 7.29 | 940 | 0.2491 | 0.7532 | 0.8228 | 0.9265 | 0.9468 | 0.4434 | 0.9097 | 0.6321 | 0.9192 | 0.9297 | 0.9784 | 0.9155 | 0.4013 | 0.7761 | 0.5376 | 0.8523 | 0.8386 | 0.9512 | 0.9254 | 4.5277 |
117
- | 0.2136 | 7.44 | 960 | 0.2462 | 0.7366 | 0.7912 | 0.9260 | 0.9747 | 0.3774 | 0.8900 | 0.5120 | 0.9313 | 0.9057 | 0.9471 | 0.9168 | 0.3526 | 0.7829 | 0.4755 | 0.8505 | 0.8361 | 0.9418 | 0.9236 | 4.5142 |
118
- | 0.2604 | 7.6 | 980 | 0.2325 | 0.7577 | 0.8278 | 0.9293 | 0.9677 | 0.4724 | 0.9051 | 0.6346 | 0.9134 | 0.9435 | 0.9581 | 0.9256 | 0.4303 | 0.7949 | 0.5457 | 0.8533 | 0.8057 | 0.9484 | 0.9282 | 4.5297 |
119
- | 0.2748 | 7.75 | 1000 | 0.2346 | 0.7698 | 0.8347 | 0.9321 | 0.9635 | 0.5114 | 0.8844 | 0.6594 | 0.9439 | 0.9167 | 0.9632 | 0.9266 | 0.4523 | 0.7917 | 0.5660 | 0.8580 | 0.8416 | 0.9527 | 0.9311 | 4.5311 |
120
- | 0.3436 | 7.91 | 1020 | 0.2411 | 0.7614 | 0.8265 | 0.9282 | 0.9724 | 0.5000 | 0.8716 | 0.6281 | 0.9297 | 0.9309 | 0.9528 | 0.9213 | 0.4477 | 0.7829 | 0.5548 | 0.8511 | 0.8259 | 0.9458 | 0.9269 | 4.5249 |
121
- | 0.2395 | 8.06 | 1040 | 0.2546 | 0.7522 | 0.8211 | 0.9264 | 0.9685 | 0.4612 | 0.9105 | 0.6059 | 0.8980 | 0.9434 | 0.9600 | 0.9245 | 0.4243 | 0.7818 | 0.5319 | 0.8413 | 0.8110 | 0.9508 | 0.9252 | 4.5256 |
122
- | 0.1813 | 8.22 | 1060 | 0.2524 | 0.7647 | 0.8297 | 0.9302 | 0.9662 | 0.5154 | 0.8952 | 0.6113 | 0.9248 | 0.9339 | 0.9609 | 0.9264 | 0.4630 | 0.7884 | 0.5406 | 0.8506 | 0.8319 | 0.9522 | 0.9292 | 4.5296 |
123
- | 0.2607 | 8.37 | 1080 | 0.2457 | 0.7770 | 0.8462 | 0.9322 | 0.9677 | 0.5659 | 0.8738 | 0.6846 | 0.9364 | 0.9318 | 0.9631 | 0.9280 | 0.4903 | 0.7938 | 0.5830 | 0.8525 | 0.8385 | 0.9531 | 0.9314 | 4.5349 |
124
- | 0.2449 | 8.53 | 1100 | 0.2444 | 0.7652 | 0.8277 | 0.9301 | 0.9723 | 0.4707 | 0.8938 | 0.6516 | 0.9228 | 0.9262 | 0.9567 | 0.9219 | 0.4271 | 0.7923 | 0.5718 | 0.8522 | 0.8426 | 0.9484 | 0.9289 | 4.5235 |
125
- | 0.2215 | 8.68 | 1120 | 0.2430 | 0.7696 | 0.8378 | 0.9301 | 0.9730 | 0.4928 | 0.8953 | 0.6959 | 0.9160 | 0.9383 | 0.9532 | 0.9223 | 0.4459 | 0.7976 | 0.5992 | 0.8523 | 0.8244 | 0.9457 | 0.9291 | 4.5243 |
126
- | 0.2453 | 8.84 | 1140 | 0.2416 | 0.7799 | 0.8507 | 0.9327 | 0.9674 | 0.6026 | 0.8718 | 0.6724 | 0.9301 | 0.9379 | 0.9728 | 0.9317 | 0.5169 | 0.7909 | 0.5879 | 0.8527 | 0.8206 | 0.9587 | 0.9321 | 4.5462 |
127
- | 0.283 | 8.99 | 1160 | 0.2409 | 0.7820 | 0.8533 | 0.9339 | 0.9646 | 0.6112 | 0.8972 | 0.6646 | 0.9200 | 0.9404 | 0.9748 | 0.9319 | 0.5263 | 0.7990 | 0.5762 | 0.8535 | 0.8275 | 0.9596 | 0.9334 | 4.5455 |
128
- | 0.226 | 9.15 | 1180 | 0.2442 | 0.7842 | 0.8568 | 0.9334 | 0.9645 | 0.6274 | 0.9011 | 0.6737 | 0.9167 | 0.9446 | 0.9694 | 0.9314 | 0.5450 | 0.8005 | 0.5831 | 0.8520 | 0.8201 | 0.9572 | 0.9330 | 4.5412 |
129
- | 0.2004 | 9.3 | 1200 | 0.2380 | 0.7802 | 0.8472 | 0.9326 | 0.9708 | 0.5978 | 0.8873 | 0.6533 | 0.9242 | 0.9333 | 0.9634 | 0.9263 | 0.5197 | 0.7991 | 0.5740 | 0.8522 | 0.8369 | 0.9532 | 0.9319 | 4.5341 |
130
- | 0.1818 | 9.46 | 1220 | 0.2384 | 0.7752 | 0.8384 | 0.9325 | 0.9713 | 0.5332 | 0.8987 | 0.6562 | 0.9231 | 0.9240 | 0.9624 | 0.9245 | 0.4765 | 0.8008 | 0.5746 | 0.8535 | 0.8443 | 0.9526 | 0.9315 | 4.5298 |
131
- | 0.2187 | 9.61 | 1240 | 0.2406 | 0.7814 | 0.8484 | 0.9334 | 0.9689 | 0.5918 | 0.8909 | 0.6568 | 0.9214 | 0.9376 | 0.9715 | 0.9284 | 0.5214 | 0.7998 | 0.5763 | 0.8529 | 0.8337 | 0.9576 | 0.9327 | 4.5414 |
132
- | 0.2394 | 9.77 | 1260 | 0.2386 | 0.7790 | 0.8458 | 0.9315 | 0.9719 | 0.5949 | 0.8797 | 0.6543 | 0.9274 | 0.9358 | 0.9567 | 0.9254 | 0.5231 | 0.7981 | 0.5779 | 0.8509 | 0.8292 | 0.9484 | 0.9308 | 4.5303 |
133
- | 0.2902 | 9.92 | 1280 | 0.2425 | 0.7753 | 0.8403 | 0.9317 | 0.9712 | 0.5496 | 0.9007 | 0.6504 | 0.9190 | 0.9343 | 0.9569 | 0.9249 | 0.4929 | 0.8001 | 0.5731 | 0.8521 | 0.8354 | 0.9485 | 0.9308 | 4.5267 |
134
 
135
 
136
  ### Framework versions
 
1
  ---
2
  license: other
3
+ base_model: nvidia/segformer-b4-finetuned-cityscapes-1024-1024
4
  tags:
5
  - vision
6
  - image-segmentation
7
  - generated_from_trainer
 
8
  model-index:
9
  - name: segformer-b4-cityscapes-finetuned-coastTrain
10
  results: []
 
17
 
18
  This model is a fine-tuned version of [nvidia/segformer-b4-finetuned-cityscapes-1024-1024](https://huggingface.co/nvidia/segformer-b4-finetuned-cityscapes-1024-1024) on the peldrak/coastTrain dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.2357
21
+ - Mean Iou: 0.7811
22
+ - Mean Accuracy: 0.8591
23
+ - Overall Accuracy: 0.9319
24
+ - Accuracy Water: 0.9549
25
+ - Accuracy Whitewater: 0.6656
26
+ - Accuracy Sediment: 0.8857
27
+ - Accuracy Other Natural Terrain: 0.7000
28
+ - Accuracy Vegetation: 0.9381
29
+ - Accuracy Development: 0.9138
30
+ - Accuracy Unknown: 0.9556
31
+ - Iou Water: 0.9109
32
+ - Iou Whitewater: 0.5055
33
+ - Iou Sediment: 0.7882
34
+ - Iou Other Natural Terrain: 0.6185
35
+ - Iou Vegetation: 0.8519
36
+ - Iou Development: 0.8427
37
+ - Iou Unknown: 0.9497
38
+ - F1 Score: 0.9319
 
39
 
40
  ## Model description
41
 
 
60
  - seed: 42
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
63
+ - num_epochs: 20
64
 
65
  ### Training results
66
 
67
+ | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Water | Accuracy Whitewater | Accuracy Sediment | Accuracy Other Natural Terrain | Accuracy Vegetation | Accuracy Development | Accuracy Unknown | Iou Water | Iou Whitewater | Iou Sediment | Iou Other Natural Terrain | Iou Vegetation | Iou Development | Iou Unknown | F1 Score |
68
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:--------------:|:-------------------:|:-----------------:|:------------------------------:|:-------------------:|:--------------------:|:----------------:|:---------:|:--------------:|:------------:|:-------------------------:|:--------------:|:---------------:|:-----------:|:--------:|
69
+ | 1.6164 | 0.16 | 20 | 1.6103 | 0.3033 | 0.4310 | 0.6173 | 0.6984 | 0.0165 | 0.1213 | 0.1453 | 0.9116 | 0.5208 | 0.6034 | 0.6076 | 0.0092 | 0.1086 | 0.0431 | 0.4403 | 0.3382 | 0.5763 | 0.6109 |
70
+ | 1.6774 | 0.31 | 40 | 1.1975 | 0.3912 | 0.5035 | 0.7266 | 0.8837 | 0.0011 | 0.2986 | 0.1865 | 0.9310 | 0.5736 | 0.6501 | 0.7156 | 0.0010 | 0.2726 | 0.1260 | 0.5542 | 0.4202 | 0.6485 | 0.7094 |
71
+ | 1.1145 | 0.47 | 60 | 0.9583 | 0.4437 | 0.5388 | 0.7842 | 0.9139 | 0.0045 | 0.5101 | 0.0021 | 0.9109 | 0.6953 | 0.7346 | 0.7547 | 0.0045 | 0.4175 | 0.0021 | 0.6288 | 0.5663 | 0.7316 | 0.7701 |
72
+ | 1.6204 | 0.62 | 80 | 0.7963 | 0.4983 | 0.5834 | 0.8301 | 0.9190 | 0.0007 | 0.7516 | 0.0 | 0.8826 | 0.6951 | 0.8351 | 0.7955 | 0.0007 | 0.5505 | 0.0 | 0.6940 | 0.6180 | 0.8292 | 0.8194 |
73
+ | 0.8495 | 0.78 | 100 | 0.6806 | 0.5207 | 0.6159 | 0.8503 | 0.9244 | 0.0000 | 0.7264 | 0.0 | 0.8833 | 0.9370 | 0.8399 | 0.8472 | 0.0000 | 0.5786 | 0.0 | 0.7045 | 0.6805 | 0.8345 | 0.8398 |
74
+ | 1.0065 | 0.93 | 120 | 0.5718 | 0.5401 | 0.6248 | 0.8639 | 0.9453 | 0.0000 | 0.7708 | 0.0 | 0.8715 | 0.9291 | 0.8567 | 0.8446 | 0.0000 | 0.6291 | 0.0 | 0.7336 | 0.7273 | 0.8459 | 0.8521 |
75
+ | 0.6766 | 1.09 | 140 | 0.5126 | 0.5445 | 0.6242 | 0.8660 | 0.9491 | 0.0 | 0.7616 | 0.0 | 0.8935 | 0.9161 | 0.8492 | 0.8401 | 0.0 | 0.6121 | 0.0 | 0.7552 | 0.7584 | 0.8457 | 0.8540 |
76
+ | 0.7925 | 1.24 | 160 | 0.5048 | 0.5329 | 0.6202 | 0.8542 | 0.9016 | 0.0003 | 0.7352 | 0.0 | 0.9269 | 0.9131 | 0.8645 | 0.8343 | 0.0003 | 0.5637 | 0.0 | 0.7175 | 0.7621 | 0.8526 | 0.8439 |
77
+ | 0.6854 | 1.4 | 180 | 0.4882 | 0.5248 | 0.6245 | 0.8503 | 0.8990 | 0.0 | 0.8316 | 0.0009 | 0.8686 | 0.9357 | 0.8356 | 0.8310 | 0.0 | 0.5829 | 0.0009 | 0.7449 | 0.6835 | 0.8308 | 0.8418 |
78
+ | 0.4662 | 1.55 | 200 | 0.4535 | 0.5386 | 0.6115 | 0.8628 | 0.9677 | 0.0 | 0.6686 | 0.0 | 0.9107 | 0.8763 | 0.8573 | 0.8473 | 0.0 | 0.5657 | 0.0 | 0.7304 | 0.7748 | 0.8518 | 0.8493 |
79
+ | 0.3362 | 1.71 | 220 | 0.4699 | 0.5325 | 0.6177 | 0.8631 | 0.9593 | 0.0000 | 0.6739 | 0.0003 | 0.9159 | 0.9334 | 0.8410 | 0.8606 | 0.0000 | 0.6050 | 0.0003 | 0.7332 | 0.6897 | 0.8385 | 0.8502 |
80
+ | 0.4076 | 1.86 | 240 | 0.4327 | 0.5554 | 0.6354 | 0.8783 | 0.9616 | 0.0 | 0.8128 | 0.0 | 0.8942 | 0.9314 | 0.8478 | 0.8682 | 0.0 | 0.6789 | 0.0 | 0.7692 | 0.7259 | 0.8456 | 0.8663 |
81
+ | 0.3913 | 2.02 | 260 | 0.3790 | 0.5601 | 0.6425 | 0.8808 | 0.9320 | 0.0024 | 0.8535 | 0.0 | 0.9221 | 0.9205 | 0.8673 | 0.8705 | 0.0024 | 0.6627 | 0.0 | 0.7890 | 0.7383 | 0.8576 | 0.8699 |
82
+ | 0.2981 | 2.17 | 280 | 0.3671 | 0.5610 | 0.6348 | 0.8824 | 0.9744 | 0.0054 | 0.7800 | 0.0 | 0.9047 | 0.9195 | 0.8599 | 0.8609 | 0.0054 | 0.6769 | 0.0 | 0.8009 | 0.7307 | 0.8525 | 0.8697 |
83
+ | 0.4318 | 2.33 | 300 | 0.3670 | 0.5621 | 0.6398 | 0.8830 | 0.9479 | 0.0020 | 0.7878 | 0.0 | 0.9324 | 0.9327 | 0.8755 | 0.8732 | 0.0020 | 0.6613 | 0.0 | 0.7800 | 0.7484 | 0.8695 | 0.8711 |
84
+ | 0.6784 | 2.48 | 320 | 0.3526 | 0.5729 | 0.6498 | 0.8880 | 0.9559 | 0.0274 | 0.8592 | 0.0013 | 0.9035 | 0.9357 | 0.8655 | 0.8731 | 0.0273 | 0.6938 | 0.0013 | 0.8037 | 0.7482 | 0.8626 | 0.8768 |
85
+ | 0.499 | 2.64 | 340 | 0.4312 | 0.5619 | 0.6425 | 0.8800 | 0.9626 | 0.0024 | 0.8855 | 0.0110 | 0.8533 | 0.9383 | 0.8446 | 0.8566 | 0.0024 | 0.6981 | 0.0110 | 0.7918 | 0.7307 | 0.8428 | 0.8687 |
86
+ | 0.8615 | 2.79 | 360 | 0.3365 | 0.5947 | 0.6622 | 0.8874 | 0.9715 | 0.0454 | 0.7257 | 0.1509 | 0.9227 | 0.9287 | 0.8906 | 0.8599 | 0.0450 | 0.6577 | 0.1487 | 0.7953 | 0.7779 | 0.8782 | 0.8773 |
87
+ | 0.3003 | 2.95 | 380 | 0.3393 | 0.6401 | 0.7163 | 0.8957 | 0.9262 | 0.1204 | 0.8909 | 0.3291 | 0.9081 | 0.9310 | 0.9085 | 0.8719 | 0.1183 | 0.6859 | 0.3225 | 0.8121 | 0.7681 | 0.9022 | 0.8915 |
88
+ | 0.5308 | 3.1 | 400 | 0.3830 | 0.5999 | 0.6674 | 0.8896 | 0.9691 | 0.0386 | 0.8396 | 0.1460 | 0.9074 | 0.9192 | 0.8520 | 0.8628 | 0.0384 | 0.6917 | 0.1454 | 0.8095 | 0.8045 | 0.8467 | 0.8804 |
89
+ | 0.282 | 3.26 | 420 | 0.3304 | 0.6647 | 0.7423 | 0.8943 | 0.9559 | 0.2274 | 0.8711 | 0.4605 | 0.8742 | 0.9358 | 0.8711 | 0.8727 | 0.2145 | 0.7110 | 0.4375 | 0.8046 | 0.7517 | 0.8611 | 0.8914 |
90
+ | 0.5241 | 3.41 | 440 | 0.3442 | 0.6755 | 0.7572 | 0.8987 | 0.9333 | 0.3115 | 0.8306 | 0.4531 | 0.9084 | 0.9409 | 0.9226 | 0.8892 | 0.2896 | 0.7121 | 0.4355 | 0.8007 | 0.7016 | 0.9001 | 0.8971 |
91
+ | 0.3216 | 3.57 | 460 | 0.3183 | 0.6791 | 0.7528 | 0.9012 | 0.9668 | 0.2982 | 0.7765 | 0.4731 | 0.9024 | 0.9413 | 0.9112 | 0.8863 | 0.2779 | 0.7006 | 0.4557 | 0.8153 | 0.7164 | 0.9016 | 0.8983 |
92
+ | 0.2215 | 3.72 | 480 | 0.3830 | 0.6360 | 0.7081 | 0.8863 | 0.9663 | 0.3204 | 0.6345 | 0.2660 | 0.9414 | 0.9120 | 0.9163 | 0.8619 | 0.2938 | 0.5889 | 0.2591 | 0.7971 | 0.7484 | 0.9032 | 0.8794 |
93
+ | 0.218 | 3.88 | 500 | 0.3043 | 0.7161 | 0.7951 | 0.9085 | 0.9319 | 0.4424 | 0.8889 | 0.5373 | 0.9031 | 0.9292 | 0.9331 | 0.8819 | 0.3956 | 0.7315 | 0.5039 | 0.8373 | 0.7575 | 0.9050 | 0.9077 |
94
+ | 1.2378 | 4.03 | 520 | 0.2986 | 0.6943 | 0.7565 | 0.9108 | 0.9551 | 0.2581 | 0.7835 | 0.4908 | 0.9377 | 0.9128 | 0.9577 | 0.8816 | 0.2385 | 0.7105 | 0.4599 | 0.8166 | 0.8091 | 0.9437 | 0.9071 |
95
+ | 0.4541 | 4.19 | 540 | 0.2573 | 0.7435 | 0.8103 | 0.9231 | 0.9569 | 0.4210 | 0.8728 | 0.6279 | 0.9147 | 0.9204 | 0.9580 | 0.8940 | 0.3789 | 0.7582 | 0.5599 | 0.8528 | 0.8206 | 0.9404 | 0.9217 |
96
+ | 0.4081 | 4.34 | 560 | 0.2838 | 0.7128 | 0.7770 | 0.9155 | 0.9546 | 0.3982 | 0.8023 | 0.4623 | 0.9336 | 0.9262 | 0.9619 | 0.8945 | 0.3503 | 0.7261 | 0.4438 | 0.8112 | 0.8117 | 0.9521 | 0.9130 |
97
+ | 0.2315 | 4.5 | 580 | 0.2708 | 0.7334 | 0.8036 | 0.9197 | 0.9704 | 0.5776 | 0.8037 | 0.4773 | 0.9332 | 0.9220 | 0.9408 | 0.9000 | 0.4597 | 0.7405 | 0.4540 | 0.8364 | 0.8102 | 0.9329 | 0.9179 |
98
+ | 0.2435 | 4.65 | 600 | 0.2466 | 0.7647 | 0.8418 | 0.9263 | 0.9479 | 0.5343 | 0.9115 | 0.7044 | 0.9117 | 0.9321 | 0.9509 | 0.9029 | 0.4430 | 0.7746 | 0.6281 | 0.8570 | 0.8114 | 0.9362 | 0.9261 |
99
+ | 0.4516 | 4.81 | 620 | 0.2416 | 0.7570 | 0.8354 | 0.9238 | 0.9662 | 0.5266 | 0.8814 | 0.7140 | 0.9014 | 0.9212 | 0.9369 | 0.9081 | 0.4453 | 0.7670 | 0.6086 | 0.8446 | 0.7996 | 0.9257 | 0.9233 |
100
+ | 0.4509 | 4.96 | 640 | 0.2508 | 0.7655 | 0.8494 | 0.9230 | 0.9551 | 0.5720 | 0.8782 | 0.7680 | 0.9144 | 0.9230 | 0.9351 | 0.8989 | 0.4836 | 0.7720 | 0.6315 | 0.8477 | 0.7990 | 0.9258 | 0.9229 |
101
+ | 0.1969 | 5.12 | 660 | 0.2465 | 0.7651 | 0.8382 | 0.9258 | 0.9530 | 0.5403 | 0.9037 | 0.6842 | 0.9110 | 0.9291 | 0.9463 | 0.9066 | 0.4604 | 0.7656 | 0.6248 | 0.8454 | 0.8118 | 0.9411 | 0.9255 |
102
+ | 0.1708 | 5.27 | 680 | 0.2227 | 0.7879 | 0.8556 | 0.9357 | 0.9711 | 0.5701 | 0.8964 | 0.7527 | 0.9016 | 0.9259 | 0.9717 | 0.9195 | 0.4858 | 0.8047 | 0.6627 | 0.8455 | 0.8394 | 0.9575 | 0.9350 |
103
+ | 1.1432 | 5.43 | 700 | 0.2491 | 0.7805 | 0.8601 | 0.9284 | 0.9502 | 0.6603 | 0.8626 | 0.7375 | 0.9469 | 0.9132 | 0.9503 | 0.9055 | 0.4984 | 0.7696 | 0.6647 | 0.8500 | 0.8299 | 0.9452 | 0.9284 |
104
+ | 0.1651 | 5.58 | 720 | 0.2540 | 0.7767 | 0.8579 | 0.9254 | 0.9510 | 0.6597 | 0.8658 | 0.7340 | 0.9306 | 0.9185 | 0.9459 | 0.8929 | 0.4721 | 0.7627 | 0.6720 | 0.8532 | 0.8441 | 0.9401 | 0.9255 |
105
+ | 0.4319 | 5.74 | 740 | 0.2351 | 0.7858 | 0.8578 | 0.9317 | 0.9647 | 0.6158 | 0.8645 | 0.7555 | 0.9330 | 0.9202 | 0.9508 | 0.9084 | 0.4928 | 0.7851 | 0.6730 | 0.8529 | 0.8415 | 0.9468 | 0.9313 |
106
+ | 0.1979 | 5.89 | 760 | 0.2278 | 0.7927 | 0.8693 | 0.9362 | 0.9554 | 0.6724 | 0.8793 | 0.7419 | 0.9298 | 0.9242 | 0.9820 | 0.9152 | 0.5258 | 0.7952 | 0.6468 | 0.8622 | 0.8458 | 0.9578 | 0.9360 |
107
+ | 0.1783 | 6.05 | 780 | 0.2357 | 0.7811 | 0.8591 | 0.9319 | 0.9549 | 0.6656 | 0.8857 | 0.7000 | 0.9381 | 0.9138 | 0.9556 | 0.9109 | 0.5055 | 0.7882 | 0.6185 | 0.8519 | 0.8427 | 0.9497 | 0.9319 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
108
 
109
 
110
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bbc757a158da73e8bd270f4b796bcc2061367f8e4ad5f25640f9093fa3a23500
3
  size 256112508
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fdde9e18ec21f547eacb4fd0c40ec67fd05349ffc4f68cbf2dcc1d0a1311392
3
  size 256112508
runs/Mar04_09-24-15_1fcd91d1db08/events.out.tfevents.1709544258.1fcd91d1db08.27.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39f6eaa3438019e7ddd69bbef7986d4d7a715a1eec547262fa8a539e19968266
3
+ size 179882
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c9a7b256fd5bd09f82881a2edc6085dc6eb2ff08a17f8f8a0e62eddb2907bbe2
3
  size 4792
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cdc6f785cbe9f5fbc7c4e601ba2a9adf8f48f7b2564ef025700a4b485aba5012
3
  size 4792