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
b1902a4
·
verified ·
1 Parent(s): c1fa091

End of training

Browse files
Files changed (4) hide show
  1. README.md +67 -78
  2. model.safetensors +1 -1
  3. runs/Mar06_13-03-38_4d04dfd18819/events.out.tfevents.1709730222.4d04dfd18819.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,25 +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.3556
21
- - Mean Iou: 0.7561
22
- - Mean Accuracy: 0.8443
23
- - Overall Accuracy: 0.9175
24
- - Accuracy Water: 0.9466
25
- - Accuracy Whitewater: 0.6901
26
- - Accuracy Sediment: 0.8516
27
- - Accuracy Other Natural Terrain: 0.6914
28
- - Accuracy Vegetation: 0.9725
29
- - Accuracy Development: 0.8959
30
- - Accuracy Unknown: 0.8623
31
- - Iou Water: 0.9040
32
- - Iou Whitewater: 0.5311
33
- - Iou Sediment: 0.7532
34
- - Iou Other Natural Terrain: 0.6223
35
- - Iou Vegetation: 0.8820
36
- - Iou Development: 0.7524
37
- - Iou Unknown: 0.8476
38
- - F1 Score: 0.9173
39
 
40
  ## Model description
41
 
@@ -66,64 +66,53 @@ The following hyperparameters were used during training:
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.6159 | 0.16 | 20 | 1.4743 | 0.3465 | 0.4471 | 0.6946 | 0.8336 | 0.0001 | 0.3708 | 0.0019 | 0.8458 | 0.4863 | 0.5910 | 0.6397 | 0.0001 | 0.2692 | 0.0016 | 0.5615 | 0.3639 | 0.5891 | 0.6796 |
70
- | 1.3039 | 0.31 | 40 | 1.1133 | 0.4229 | 0.5298 | 0.7670 | 0.8703 | 0.0000 | 0.6398 | 0.0000 | 0.8251 | 0.6148 | 0.7589 | 0.7303 | 0.0000 | 0.3461 | 0.0000 | 0.6582 | 0.4702 | 0.7556 | 0.7615 |
71
- | 1.1378 | 0.47 | 60 | 0.9002 | 0.4549 | 0.5455 | 0.8106 | 0.8950 | 0.0001 | 0.4156 | 0.0000 | 0.9384 | 0.7718 | 0.7978 | 0.7862 | 0.0001 | 0.3617 | 0.0000 | 0.7144 | 0.5295 | 0.7920 | 0.7949 |
72
- | 0.8296 | 0.62 | 80 | 0.7117 | 0.4853 | 0.5698 | 0.8343 | 0.9301 | 0.0 | 0.6730 | 0.0 | 0.9536 | 0.6391 | 0.7927 | 0.8103 | 0.0 | 0.5208 | 0.0 | 0.7637 | 0.5108 | 0.7911 | 0.8202 |
73
- | 1.4277 | 0.78 | 100 | 0.6559 | 0.4795 | 0.5637 | 0.8313 | 0.9643 | 0.0 | 0.5861 | 0.0 | 0.9123 | 0.6822 | 0.8008 | 0.7803 | 0.0 | 0.4585 | 0.0 | 0.7906 | 0.5310 | 0.7958 | 0.8168 |
74
- | 0.915 | 0.93 | 120 | 0.6692 | 0.4593 | 0.5696 | 0.8003 | 0.9383 | 0.0 | 0.8360 | 0.0 | 0.7728 | 0.6365 | 0.8035 | 0.7894 | 0.0 | 0.3948 | 0.0 | 0.7080 | 0.5249 | 0.7978 | 0.7975 |
75
- | 0.5845 | 1.09 | 140 | 0.5152 | 0.4939 | 0.5883 | 0.8398 | 0.9255 | 0.0 | 0.7697 | 0.0 | 0.9243 | 0.6947 | 0.8040 | 0.8302 | 0.0 | 0.4814 | 0.0 | 0.7844 | 0.5639 | 0.7976 | 0.8298 |
76
- | 0.6603 | 1.24 | 160 | 0.4796 | 0.5211 | 0.6072 | 0.8577 | 0.9726 | 0.0 | 0.7405 | 0.0 | 0.9000 | 0.8342 | 0.8029 | 0.8280 | 0.0 | 0.5529 | 0.0 | 0.8020 | 0.6675 | 0.7974 | 0.8450 |
77
- | 1.0074 | 1.4 | 180 | 0.4792 | 0.5257 | 0.6158 | 0.8621 | 0.9271 | 0.0 | 0.7763 | 0.0 | 0.9451 | 0.8562 | 0.8061 | 0.8431 | 0.0 | 0.5783 | 0.0 | 0.8138 | 0.6467 | 0.7984 | 0.8503 |
78
- | 0.3431 | 1.55 | 200 | 0.4631 | 0.5344 | 0.6251 | 0.8682 | 0.9492 | 0.0 | 0.8284 | 0.0 | 0.9232 | 0.8698 | 0.8053 | 0.8460 | 0.0 | 0.6038 | 0.0 | 0.8304 | 0.6619 | 0.7988 | 0.8565 |
79
- | 0.3317 | 1.71 | 220 | 0.4442 | 0.5296 | 0.6087 | 0.8641 | 0.9459 | 0.0000 | 0.7155 | 0.0 | 0.9544 | 0.8380 | 0.8069 | 0.8510 | 0.0000 | 0.5947 | 0.0 | 0.7913 | 0.6723 | 0.7981 | 0.8505 |
80
- | 0.7125 | 1.86 | 240 | 0.4652 | 0.5287 | 0.6123 | 0.8609 | 0.9455 | 0.0000 | 0.7733 | 0.0 | 0.9300 | 0.8316 | 0.8057 | 0.8392 | 0.0000 | 0.6147 | 0.0 | 0.7861 | 0.6630 | 0.7979 | 0.8481 |
81
- | 0.2776 | 2.02 | 260 | 0.4849 | 0.5259 | 0.6016 | 0.8644 | 0.9598 | 0.0000 | 0.6631 | 0.0 | 0.9632 | 0.8203 | 0.8044 | 0.8497 | 0.0000 | 0.5649 | 0.0 | 0.8021 | 0.6651 | 0.7992 | 0.8499 |
82
- | 0.7679 | 2.17 | 280 | 0.4463 | 0.5333 | 0.6188 | 0.8653 | 0.9482 | 0.0010 | 0.7903 | 0.0 | 0.9284 | 0.8573 | 0.8065 | 0.8407 | 0.0010 | 0.5719 | 0.0 | 0.8101 | 0.7100 | 0.7996 | 0.8531 |
83
- | 0.571 | 2.33 | 300 | 0.4167 | 0.5440 | 0.6279 | 0.8731 | 0.9367 | 0.0173 | 0.8056 | 0.0065 | 0.9581 | 0.8590 | 0.8120 | 0.8664 | 0.0173 | 0.6075 | 0.0065 | 0.8185 | 0.6910 | 0.8006 | 0.8612 |
84
- | 0.451 | 2.48 | 320 | 0.4462 | 0.5393 | 0.6214 | 0.8734 | 0.9691 | 0.0046 | 0.7837 | 0.0022 | 0.9360 | 0.8399 | 0.8143 | 0.8518 | 0.0045 | 0.5904 | 0.0022 | 0.8424 | 0.6799 | 0.8038 | 0.8607 |
85
- | 0.4442 | 2.64 | 340 | 0.4778 | 0.5263 | 0.6287 | 0.8555 | 0.9441 | 0.0371 | 0.8508 | 0.0 | 0.8623 | 0.8909 | 0.8158 | 0.8669 | 0.0367 | 0.5251 | 0.0 | 0.7737 | 0.6770 | 0.8048 | 0.8477 |
86
- | 0.3257 | 2.79 | 360 | 0.4507 | 0.5432 | 0.6384 | 0.8628 | 0.9527 | 0.0327 | 0.8440 | 0.0508 | 0.8735 | 0.8931 | 0.8220 | 0.8595 | 0.0324 | 0.5533 | 0.0502 | 0.7834 | 0.7140 | 0.8098 | 0.8542 |
87
- | 0.7477 | 2.95 | 380 | 0.3665 | 0.5797 | 0.6446 | 0.8904 | 0.9627 | 0.0750 | 0.7582 | 0.0186 | 0.9653 | 0.8778 | 0.8544 | 0.8720 | 0.0731 | 0.6663 | 0.0186 | 0.8306 | 0.7623 | 0.8350 | 0.8785 |
88
- | 0.2692 | 3.1 | 400 | 0.4098 | 0.5707 | 0.6486 | 0.8840 | 0.9536 | 0.1067 | 0.8264 | 0.0026 | 0.9564 | 0.8753 | 0.8192 | 0.8729 | 0.1035 | 0.6058 | 0.0025 | 0.8395 | 0.7593 | 0.8112 | 0.8738 |
89
- | 0.4504 | 3.26 | 420 | 0.3821 | 0.5977 | 0.6698 | 0.8874 | 0.9631 | 0.2230 | 0.7329 | 0.0862 | 0.9539 | 0.8833 | 0.8462 | 0.8712 | 0.2002 | 0.6279 | 0.0857 | 0.8354 | 0.7374 | 0.8262 | 0.8791 |
90
- | 0.426 | 3.41 | 440 | 0.3995 | 0.5910 | 0.6636 | 0.8856 | 0.9655 | 0.2359 | 0.8131 | 0.0085 | 0.9567 | 0.8530 | 0.8127 | 0.8691 | 0.2182 | 0.6564 | 0.0085 | 0.8327 | 0.7526 | 0.7993 | 0.8764 |
91
- | 0.5443 | 3.57 | 460 | 0.3611 | 0.6173 | 0.6880 | 0.8991 | 0.9471 | 0.2484 | 0.8521 | 0.0488 | 0.9532 | 0.8842 | 0.8819 | 0.8751 | 0.2234 | 0.6843 | 0.0481 | 0.8502 | 0.7845 | 0.8555 | 0.8916 |
92
- | 0.4419 | 3.72 | 480 | 0.3920 | 0.6111 | 0.6927 | 0.8876 | 0.9396 | 0.3563 | 0.8102 | 0.0611 | 0.9618 | 0.8956 | 0.8243 | 0.8837 | 0.3152 | 0.6403 | 0.0608 | 0.8305 | 0.7323 | 0.8149 | 0.8815 |
93
- | 0.4769 | 3.88 | 500 | 0.3631 | 0.6276 | 0.7045 | 0.8990 | 0.9527 | 0.3812 | 0.8409 | 0.0492 | 0.9512 | 0.8902 | 0.8658 | 0.8872 | 0.3400 | 0.6772 | 0.0490 | 0.8594 | 0.7429 | 0.8373 | 0.8927 |
94
- | 0.3016 | 4.03 | 520 | 0.3577 | 0.6473 | 0.7262 | 0.8989 | 0.9493 | 0.4276 | 0.8062 | 0.1820 | 0.9519 | 0.8992 | 0.8675 | 0.8857 | 0.3481 | 0.6790 | 0.1769 | 0.8505 | 0.7499 | 0.8407 | 0.8947 |
95
- | 0.3139 | 4.19 | 540 | 0.3647 | 0.6394 | 0.7213 | 0.8980 | 0.9615 | 0.4900 | 0.7992 | 0.1045 | 0.9456 | 0.8868 | 0.8615 | 0.8899 | 0.3818 | 0.6598 | 0.1023 | 0.8494 | 0.7541 | 0.8384 | 0.8932 |
96
- | 0.7716 | 4.34 | 560 | 0.3705 | 0.6355 | 0.7131 | 0.8977 | 0.9637 | 0.4701 | 0.7886 | 0.0669 | 0.9528 | 0.9031 | 0.8463 | 0.8825 | 0.3893 | 0.6725 | 0.0654 | 0.8541 | 0.7561 | 0.8282 | 0.8917 |
97
- | 0.2064 | 4.5 | 580 | 0.3519 | 0.6475 | 0.7311 | 0.9003 | 0.9550 | 0.4379 | 0.8497 | 0.1562 | 0.9391 | 0.9258 | 0.8543 | 0.8795 | 0.3560 | 0.6969 | 0.1517 | 0.8697 | 0.7477 | 0.8311 | 0.8963 |
98
- | 0.328 | 4.65 | 600 | 0.3472 | 0.6668 | 0.7453 | 0.9030 | 0.9492 | 0.5101 | 0.8296 | 0.2324 | 0.9661 | 0.8668 | 0.8627 | 0.8829 | 0.3997 | 0.6923 | 0.2227 | 0.8630 | 0.7680 | 0.8389 | 0.8997 |
99
- | 0.232 | 4.81 | 620 | 0.3783 | 0.7012 | 0.8077 | 0.8943 | 0.9511 | 0.5130 | 0.8176 | 0.7141 | 0.9107 | 0.8877 | 0.8599 | 0.8866 | 0.3894 | 0.6315 | 0.5600 | 0.8175 | 0.7807 | 0.8426 | 0.8945 |
100
- | 0.2855 | 4.96 | 640 | 0.3812 | 0.6955 | 0.7845 | 0.9027 | 0.9371 | 0.5656 | 0.8306 | 0.4463 | 0.9620 | 0.8968 | 0.8529 | 0.8823 | 0.4207 | 0.6727 | 0.4128 | 0.8548 | 0.7783 | 0.8470 | 0.9018 |
101
- | 0.2412 | 5.12 | 660 | 0.3655 | 0.7137 | 0.7974 | 0.9056 | 0.9459 | 0.5650 | 0.8384 | 0.5349 | 0.9625 | 0.8866 | 0.8485 | 0.8817 | 0.4150 | 0.7214 | 0.5173 | 0.8632 | 0.7662 | 0.8312 | 0.9046 |
102
- | 0.2082 | 5.27 | 680 | 0.3343 | 0.7154 | 0.7934 | 0.9107 | 0.9527 | 0.5929 | 0.8280 | 0.4559 | 0.9642 | 0.8966 | 0.8639 | 0.8959 | 0.4573 | 0.7323 | 0.4429 | 0.8625 | 0.7733 | 0.8435 | 0.9092 |
103
- | 0.4416 | 5.43 | 700 | 0.3514 | 0.7158 | 0.8003 | 0.9106 | 0.9331 | 0.6079 | 0.8660 | 0.4651 | 0.9703 | 0.8886 | 0.8712 | 0.8901 | 0.4638 | 0.7276 | 0.4346 | 0.8657 | 0.7771 | 0.8519 | 0.9096 |
104
- | 0.191 | 5.58 | 720 | 0.3179 | 0.7306 | 0.8196 | 0.9125 | 0.9549 | 0.6056 | 0.8420 | 0.6048 | 0.9500 | 0.9155 | 0.8646 | 0.8965 | 0.4449 | 0.7542 | 0.5473 | 0.8730 | 0.7584 | 0.8402 | 0.9120 |
105
- | 0.1884 | 5.74 | 740 | 0.3441 | 0.7366 | 0.8194 | 0.9126 | 0.9441 | 0.5602 | 0.8682 | 0.6389 | 0.9625 | 0.9001 | 0.8616 | 0.8940 | 0.4449 | 0.7405 | 0.5858 | 0.8699 | 0.7836 | 0.8373 | 0.9118 |
106
- | 0.277 | 5.89 | 760 | 0.3121 | 0.7419 | 0.8205 | 0.9179 | 0.9577 | 0.6134 | 0.8338 | 0.5970 | 0.9604 | 0.8962 | 0.8852 | 0.8999 | 0.4515 | 0.7449 | 0.5659 | 0.8772 | 0.7983 | 0.8554 | 0.9171 |
107
- | 0.181 | 6.05 | 780 | 0.3543 | 0.7314 | 0.8129 | 0.9141 | 0.9517 | 0.6239 | 0.8661 | 0.5168 | 0.9583 | 0.9120 | 0.8611 | 0.8967 | 0.4857 | 0.7511 | 0.4988 | 0.8758 | 0.7719 | 0.8397 | 0.9131 |
108
- | 0.1812 | 6.2 | 800 | 0.3260 | 0.7370 | 0.8187 | 0.9140 | 0.9581 | 0.5996 | 0.8174 | 0.6378 | 0.9665 | 0.8854 | 0.8660 | 0.9089 | 0.4675 | 0.7414 | 0.5870 | 0.8731 | 0.7410 | 0.8404 | 0.9132 |
109
- | 0.1737 | 6.36 | 820 | 0.3332 | 0.7311 | 0.8173 | 0.9134 | 0.9587 | 0.6011 | 0.8190 | 0.6142 | 0.9651 | 0.9095 | 0.8536 | 0.9089 | 0.4873 | 0.7495 | 0.5235 | 0.8716 | 0.7367 | 0.8402 | 0.9127 |
110
- | 0.2288 | 6.51 | 840 | 0.3547 | 0.7397 | 0.8353 | 0.9146 | 0.9514 | 0.6385 | 0.8815 | 0.6453 | 0.9457 | 0.9239 | 0.8611 | 0.9052 | 0.5085 | 0.7485 | 0.5461 | 0.8735 | 0.7437 | 0.8528 | 0.9148 |
111
- | 0.2722 | 6.67 | 860 | 0.3257 | 0.7501 | 0.8314 | 0.9175 | 0.9495 | 0.6057 | 0.8482 | 0.6794 | 0.9728 | 0.8990 | 0.8656 | 0.9072 | 0.4650 | 0.7707 | 0.6232 | 0.8764 | 0.7683 | 0.8397 | 0.9169 |
112
- | 0.2327 | 6.82 | 880 | 0.3177 | 0.7464 | 0.8279 | 0.9170 | 0.9540 | 0.6026 | 0.8618 | 0.6429 | 0.9603 | 0.9067 | 0.8672 | 0.9030 | 0.4690 | 0.7625 | 0.5943 | 0.8762 | 0.7777 | 0.8421 | 0.9163 |
113
- | 0.247 | 6.98 | 900 | 0.3146 | 0.7526 | 0.8490 | 0.9174 | 0.9402 | 0.6725 | 0.8768 | 0.7106 | 0.9599 | 0.9089 | 0.8740 | 0.9039 | 0.4700 | 0.7599 | 0.6354 | 0.8837 | 0.7748 | 0.8408 | 0.9176 |
114
- | 0.2542 | 7.13 | 920 | 0.3387 | 0.7374 | 0.8389 | 0.9121 | 0.9573 | 0.6641 | 0.8635 | 0.6808 | 0.9370 | 0.9068 | 0.8626 | 0.9174 | 0.5178 | 0.7097 | 0.5533 | 0.8533 | 0.7567 | 0.8537 | 0.9124 |
115
- | 0.1622 | 7.29 | 940 | 0.3292 | 0.7482 | 0.8385 | 0.9186 | 0.9622 | 0.6632 | 0.8540 | 0.6643 | 0.9612 | 0.9071 | 0.8574 | 0.9170 | 0.5046 | 0.7675 | 0.5736 | 0.8747 | 0.7474 | 0.8521 | 0.9184 |
116
- | 0.1368 | 7.44 | 960 | 0.3336 | 0.7415 | 0.8419 | 0.9151 | 0.9474 | 0.6611 | 0.8813 | 0.6698 | 0.9465 | 0.9200 | 0.8669 | 0.9064 | 0.4912 | 0.7463 | 0.5738 | 0.8774 | 0.7416 | 0.8540 | 0.9155 |
117
- | 0.2087 | 7.6 | 980 | 0.3307 | 0.7536 | 0.8373 | 0.9185 | 0.9561 | 0.6598 | 0.8369 | 0.6690 | 0.9684 | 0.9071 | 0.8639 | 0.9108 | 0.4946 | 0.7646 | 0.6196 | 0.8785 | 0.7635 | 0.8437 | 0.9181 |
118
- | 0.2024 | 7.75 | 1000 | 0.3194 | 0.7605 | 0.8388 | 0.9201 | 0.9688 | 0.6495 | 0.8571 | 0.6775 | 0.9608 | 0.9010 | 0.8572 | 0.9036 | 0.4891 | 0.7906 | 0.6303 | 0.8787 | 0.7939 | 0.8374 | 0.9194 |
119
- | 0.2944 | 7.91 | 1020 | 0.3314 | 0.7541 | 0.8434 | 0.9175 | 0.9483 | 0.6595 | 0.8877 | 0.6750 | 0.9554 | 0.9124 | 0.8656 | 0.8965 | 0.4958 | 0.7686 | 0.6094 | 0.8814 | 0.7844 | 0.8423 | 0.9173 |
120
- | 0.2359 | 8.06 | 1040 | 0.3369 | 0.7373 | 0.8209 | 0.9143 | 0.9532 | 0.6466 | 0.8037 | 0.6139 | 0.9730 | 0.8879 | 0.8680 | 0.9019 | 0.4558 | 0.7295 | 0.5742 | 0.8700 | 0.7864 | 0.8435 | 0.9135 |
121
- | 0.3261 | 8.22 | 1060 | 0.3308 | 0.7565 | 0.8404 | 0.9195 | 0.9564 | 0.6398 | 0.8636 | 0.6987 | 0.9672 | 0.8914 | 0.8656 | 0.9047 | 0.4834 | 0.7708 | 0.6200 | 0.8783 | 0.7963 | 0.8422 | 0.9189 |
122
- | 0.1086 | 8.37 | 1080 | 0.3293 | 0.7539 | 0.8386 | 0.9189 | 0.9516 | 0.6520 | 0.8648 | 0.6747 | 0.9675 | 0.8878 | 0.8716 | 0.9034 | 0.4811 | 0.7606 | 0.6132 | 0.8798 | 0.7950 | 0.8443 | 0.9185 |
123
- | 0.2145 | 8.53 | 1100 | 0.3120 | 0.7568 | 0.8356 | 0.9211 | 0.9515 | 0.6277 | 0.8697 | 0.6506 | 0.9684 | 0.9067 | 0.8745 | 0.9080 | 0.4859 | 0.7750 | 0.6132 | 0.8841 | 0.7839 | 0.8478 | 0.9206 |
124
- | 0.2278 | 8.68 | 1120 | 0.3407 | 0.7588 | 0.8470 | 0.9194 | 0.9553 | 0.7077 | 0.8781 | 0.6847 | 0.9701 | 0.8735 | 0.8599 | 0.9094 | 0.5247 | 0.7740 | 0.6140 | 0.8786 | 0.7664 | 0.8448 | 0.9191 |
125
- | 0.983 | 8.84 | 1140 | 0.3173 | 0.7529 | 0.8349 | 0.9193 | 0.9628 | 0.6234 | 0.8341 | 0.6921 | 0.9652 | 0.8985 | 0.8684 | 0.9098 | 0.4725 | 0.7579 | 0.6328 | 0.8825 | 0.7686 | 0.8464 | 0.9186 |
126
- | 0.0743 | 8.99 | 1160 | 0.3556 | 0.7561 | 0.8443 | 0.9175 | 0.9466 | 0.6901 | 0.8516 | 0.6914 | 0.9725 | 0.8959 | 0.8623 | 0.9040 | 0.5311 | 0.7532 | 0.6223 | 0.8820 | 0.7524 | 0.8476 | 0.9173 |
127
 
128
 
129
  ### 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.2345
21
+ - Mean Iou: 0.7920
22
+ - Mean Accuracy: 0.8609
23
+ - Overall Accuracy: 0.9360
24
+ - Accuracy Water: 0.9603
25
+ - Accuracy Whitewater: 0.6503
26
+ - Accuracy Sediment: 0.8872
27
+ - Accuracy Other Natural Terrain: 0.6902
28
+ - Accuracy Vegetation: 0.9383
29
+ - Accuracy Development: 0.9340
30
+ - Accuracy Unknown: 0.9659
31
+ - Iou Water: 0.9194
32
+ - Iou Whitewater: 0.5196
33
+ - Iou Sediment: 0.8231
34
+ - Iou Other Natural Terrain: 0.6344
35
+ - Iou Vegetation: 0.8728
36
+ - Iou Development: 0.8571
37
+ - Iou Unknown: 0.9179
38
+ - F1 Score: 0.9354
39
 
40
  ## Model description
41
 
 
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.6229 | 0.16 | 20 | 1.5080 | 0.3461 | 0.4464 | 0.6629 | 0.7064 | 0.0117 | 0.3677 | 0.0094 | 0.9033 | 0.4311 | 0.6952 | 0.6022 | 0.0115 | 0.2505 | 0.0054 | 0.5058 | 0.3657 | 0.6818 | 0.6578 |
70
+ | 1.6972 | 0.31 | 40 | 1.1157 | 0.4168 | 0.5175 | 0.7412 | 0.7367 | 0.0 | 0.5983 | 0.0000 | 0.9738 | 0.5606 | 0.7528 | 0.6767 | 0.0 | 0.4343 | 0.0000 | 0.5731 | 0.4817 | 0.7519 | 0.7355 |
71
+ | 1.1969 | 0.47 | 60 | 0.8521 | 0.4777 | 0.5606 | 0.8157 | 0.8778 | 0.0001 | 0.5850 | 0.0000 | 0.9674 | 0.5962 | 0.8978 | 0.7865 | 0.0001 | 0.4970 | 0.0000 | 0.6699 | 0.4942 | 0.8960 | 0.8025 |
72
+ | 1.2209 | 0.62 | 80 | 0.6990 | 0.5024 | 0.6009 | 0.8277 | 0.8306 | 0.0 | 0.7406 | 0.0 | 0.9174 | 0.7970 | 0.9208 | 0.7942 | 0.0 | 0.5641 | 0.0 | 0.6800 | 0.5612 | 0.9175 | 0.8201 |
73
+ | 0.793 | 0.78 | 100 | 0.5543 | 0.5256 | 0.6016 | 0.8590 | 0.9409 | 0.0 | 0.6744 | 0.0 | 0.9365 | 0.7280 | 0.9312 | 0.8480 | 0.0 | 0.5903 | 0.0 | 0.7405 | 0.5737 | 0.9264 | 0.8458 |
74
+ | 1.1597 | 0.93 | 120 | 0.4957 | 0.5524 | 0.6231 | 0.8720 | 0.9201 | 0.0 | 0.7323 | 0.0 | 0.9560 | 0.8134 | 0.9401 | 0.8588 | 0.0 | 0.6428 | 0.0 | 0.7344 | 0.6976 | 0.9335 | 0.8602 |
75
+ | 0.689 | 1.09 | 140 | 0.4376 | 0.5692 | 0.6374 | 0.8844 | 0.9221 | 0.0 | 0.7930 | 0.0000 | 0.9586 | 0.8469 | 0.9411 | 0.8748 | 0.0 | 0.6899 | 0.0000 | 0.7542 | 0.7310 | 0.9348 | 0.8730 |
76
+ | 0.9423 | 1.24 | 160 | 0.3968 | 0.5715 | 0.6433 | 0.8882 | 0.9328 | 0.0000 | 0.8273 | 0.0 | 0.9201 | 0.8664 | 0.9564 | 0.8794 | 0.0000 | 0.6963 | 0.0 | 0.7687 | 0.7135 | 0.9424 | 0.8766 |
77
+ | 1.2176 | 1.4 | 180 | 0.3838 | 0.5673 | 0.6397 | 0.8811 | 0.9107 | 0.0000 | 0.7816 | 0.0 | 0.9461 | 0.8832 | 0.9561 | 0.8578 | 0.0000 | 0.6427 | 0.0 | 0.7728 | 0.7550 | 0.9429 | 0.8695 |
78
+ | 0.4714 | 1.55 | 200 | 0.4459 | 0.5380 | 0.6092 | 0.8570 | 0.9730 | 0.0 | 0.5500 | 0.0 | 0.8806 | 0.9121 | 0.9490 | 0.7998 | 0.0 | 0.4883 | 0.0 | 0.7722 | 0.7618 | 0.9442 | 0.8398 |
79
+ | 0.5087 | 1.71 | 220 | 0.4062 | 0.5677 | 0.6359 | 0.8827 | 0.9365 | 0.0008 | 0.7216 | 0.0 | 0.9511 | 0.8918 | 0.9499 | 0.8844 | 0.0008 | 0.6722 | 0.0 | 0.7302 | 0.7463 | 0.9402 | 0.8709 |
80
+ | 0.484 | 1.86 | 240 | 0.3121 | 0.5926 | 0.6518 | 0.9017 | 0.9688 | 0.0001 | 0.7858 | 0.0 | 0.9341 | 0.9243 | 0.9498 | 0.8788 | 0.0001 | 0.7236 | 0.0 | 0.8069 | 0.7972 | 0.9420 | 0.8886 |
81
+ | 0.443 | 2.02 | 260 | 0.3554 | 0.5811 | 0.6575 | 0.8904 | 0.8939 | 0.0001 | 0.8842 | 0.0 | 0.9316 | 0.9332 | 0.9599 | 0.8541 | 0.0001 | 0.6697 | 0.0 | 0.8205 | 0.7724 | 0.9512 | 0.8798 |
82
+ | 0.466 | 2.17 | 280 | 0.3265 | 0.5830 | 0.6553 | 0.8954 | 0.9347 | 0.0021 | 0.8131 | 0.0 | 0.9256 | 0.9487 | 0.9627 | 0.8786 | 0.0021 | 0.6993 | 0.0 | 0.7950 | 0.7566 | 0.9494 | 0.8833 |
83
+ | 0.7117 | 2.33 | 300 | 0.4096 | 0.5634 | 0.6494 | 0.8672 | 0.8367 | 0.0076 | 0.9133 | 0.0 | 0.9048 | 0.9233 | 0.9600 | 0.8000 | 0.0076 | 0.5985 | 0.0 | 0.8080 | 0.7798 | 0.9503 | 0.8595 |
84
+ | 0.3095 | 2.48 | 320 | 0.3111 | 0.5858 | 0.6494 | 0.8986 | 0.9655 | 0.0070 | 0.7517 | 0.0 | 0.9363 | 0.9210 | 0.9644 | 0.8850 | 0.0070 | 0.6808 | 0.0 | 0.8089 | 0.7790 | 0.9399 | 0.8851 |
85
+ | 0.3843 | 2.64 | 340 | 0.3076 | 0.6033 | 0.6589 | 0.9050 | 0.9648 | 0.0448 | 0.7872 | 0.0 | 0.9478 | 0.9052 | 0.9628 | 0.8786 | 0.0443 | 0.6973 | 0.0 | 0.8322 | 0.8187 | 0.9516 | 0.8924 |
86
+ | 0.4158 | 2.79 | 360 | 0.2985 | 0.5965 | 0.6553 | 0.9026 | 0.9557 | 0.0266 | 0.8130 | 0.0 | 0.9561 | 0.8939 | 0.9415 | 0.8820 | 0.0265 | 0.7245 | 0.0 | 0.8147 | 0.7892 | 0.9386 | 0.8903 |
87
+ | 0.3492 | 2.95 | 380 | 0.2709 | 0.6251 | 0.6863 | 0.9126 | 0.9524 | 0.1545 | 0.8818 | 0.0 | 0.9347 | 0.9218 | 0.9587 | 0.9003 | 0.1514 | 0.7515 | 0.0 | 0.8313 | 0.7895 | 0.9520 | 0.9026 |
88
+ | 0.2384 | 3.1 | 400 | 0.2531 | 0.6348 | 0.6950 | 0.9169 | 0.9541 | 0.1876 | 0.8731 | 0.0015 | 0.9446 | 0.9407 | 0.9632 | 0.9061 | 0.1830 | 0.7663 | 0.0015 | 0.8395 | 0.7944 | 0.9524 | 0.9071 |
89
+ | 0.2227 | 3.26 | 420 | 0.2772 | 0.6388 | 0.6939 | 0.9186 | 0.9648 | 0.1767 | 0.9028 | 0.0013 | 0.9230 | 0.9247 | 0.9637 | 0.9015 | 0.1735 | 0.7597 | 0.0013 | 0.8503 | 0.8312 | 0.9544 | 0.9087 |
90
+ | 0.6677 | 3.41 | 440 | 0.2861 | 0.6306 | 0.6874 | 0.9091 | 0.9640 | 0.1876 | 0.7780 | 0.0559 | 0.9635 | 0.9022 | 0.9610 | 0.8965 | 0.1838 | 0.7228 | 0.0558 | 0.8215 | 0.7825 | 0.9509 | 0.8996 |
91
+ | 0.3552 | 3.57 | 460 | 0.2795 | 0.6408 | 0.6945 | 0.9130 | 0.9641 | 0.2606 | 0.8234 | 0.0001 | 0.9487 | 0.8958 | 0.9690 | 0.8932 | 0.2536 | 0.7201 | 0.0001 | 0.8427 | 0.8204 | 0.9553 | 0.9032 |
92
+ | 0.3258 | 3.72 | 480 | 0.3075 | 0.6306 | 0.6891 | 0.9042 | 0.9730 | 0.2962 | 0.7091 | 0.0068 | 0.9628 | 0.9131 | 0.9625 | 0.8945 | 0.2617 | 0.6816 | 0.0068 | 0.8003 | 0.8150 | 0.9546 | 0.8939 |
93
+ | 0.4778 | 3.88 | 500 | 0.2449 | 0.6570 | 0.7137 | 0.9188 | 0.9583 | 0.2928 | 0.8807 | 0.0347 | 0.9383 | 0.9280 | 0.9633 | 0.9028 | 0.2768 | 0.7677 | 0.0347 | 0.8407 | 0.8223 | 0.9541 | 0.9106 |
94
+ | 0.4817 | 4.03 | 520 | 0.2365 | 0.6790 | 0.7381 | 0.9235 | 0.9611 | 0.4327 | 0.8871 | 0.0484 | 0.9393 | 0.9328 | 0.9651 | 0.9121 | 0.3866 | 0.7842 | 0.0483 | 0.8459 | 0.8230 | 0.9529 | 0.9165 |
95
+ | 0.3363 | 4.19 | 540 | 0.2273 | 0.6783 | 0.7315 | 0.9243 | 0.9635 | 0.4529 | 0.8805 | 0.0058 | 0.9546 | 0.8945 | 0.9685 | 0.9131 | 0.4101 | 0.7847 | 0.0058 | 0.8451 | 0.8339 | 0.9554 | 0.9163 |
96
+ | 0.4825 | 4.34 | 560 | 0.2615 | 0.6791 | 0.7482 | 0.9180 | 0.9406 | 0.5008 | 0.9124 | 0.0441 | 0.9206 | 0.9512 | 0.9675 | 0.8996 | 0.4385 | 0.7400 | 0.0441 | 0.8552 | 0.8198 | 0.9562 | 0.9119 |
97
+ | 0.3482 | 4.5 | 580 | 0.2336 | 0.6965 | 0.7537 | 0.9276 | 0.9695 | 0.4845 | 0.8611 | 0.1015 | 0.9492 | 0.9455 | 0.9648 | 0.9193 | 0.4258 | 0.7901 | 0.1006 | 0.8498 | 0.8349 | 0.9550 | 0.9215 |
98
+ | 0.5311 | 4.65 | 600 | 0.2592 | 0.6858 | 0.7484 | 0.9200 | 0.9477 | 0.4867 | 0.8797 | 0.0974 | 0.9463 | 0.9109 | 0.9703 | 0.9136 | 0.4329 | 0.7687 | 0.0970 | 0.8438 | 0.8232 | 0.9215 | 0.9142 |
99
+ | 0.3754 | 4.81 | 620 | 0.2345 | 0.7039 | 0.7629 | 0.9265 | 0.9641 | 0.5201 | 0.8692 | 0.1376 | 0.9387 | 0.9338 | 0.9769 | 0.9233 | 0.4557 | 0.7998 | 0.1369 | 0.8395 | 0.8404 | 0.9315 | 0.9211 |
100
+ | 0.236 | 4.96 | 640 | 0.2342 | 0.7061 | 0.7604 | 0.9268 | 0.9669 | 0.4754 | 0.8856 | 0.1790 | 0.9329 | 0.9039 | 0.9790 | 0.9218 | 0.4330 | 0.8088 | 0.1763 | 0.8374 | 0.8329 | 0.9327 | 0.9218 |
101
+ | 0.3496 | 5.12 | 660 | 0.2061 | 0.7264 | 0.7870 | 0.9313 | 0.9622 | 0.5779 | 0.9263 | 0.2075 | 0.9189 | 0.9404 | 0.9757 | 0.9243 | 0.5092 | 0.8247 | 0.2026 | 0.8525 | 0.8402 | 0.9315 | 0.9273 |
102
+ | 0.1729 | 5.27 | 680 | 0.2289 | 0.7086 | 0.7793 | 0.9245 | 0.9541 | 0.6032 | 0.8708 | 0.1744 | 0.9351 | 0.9392 | 0.9781 | 0.9202 | 0.4789 | 0.7981 | 0.1706 | 0.8377 | 0.8265 | 0.9282 | 0.9203 |
103
+ | 0.2636 | 5.43 | 700 | 0.2071 | 0.7739 | 0.8389 | 0.9335 | 0.9623 | 0.6448 | 0.8970 | 0.5348 | 0.9262 | 0.9396 | 0.9676 | 0.9245 | 0.5537 | 0.8125 | 0.4941 | 0.8551 | 0.8473 | 0.9304 | 0.9324 |
104
+ | 0.1594 | 5.58 | 720 | 0.2175 | 0.7447 | 0.8114 | 0.9284 | 0.9632 | 0.6275 | 0.8570 | 0.3850 | 0.9306 | 0.9383 | 0.9783 | 0.9172 | 0.5130 | 0.7999 | 0.3765 | 0.8467 | 0.8207 | 0.9389 | 0.9262 |
105
+ | 0.6799 | 5.74 | 740 | 0.1965 | 0.7704 | 0.8330 | 0.9379 | 0.9650 | 0.6576 | 0.9113 | 0.4469 | 0.9289 | 0.9430 | 0.9783 | 0.9303 | 0.5574 | 0.8353 | 0.4280 | 0.8695 | 0.8384 | 0.9338 | 0.9364 |
106
+ | 0.4955 | 5.89 | 760 | 0.2184 | 0.7480 | 0.8065 | 0.9318 | 0.9569 | 0.5099 | 0.9085 | 0.4344 | 0.9266 | 0.9255 | 0.9841 | 0.9146 | 0.4472 | 0.8109 | 0.4160 | 0.8654 | 0.8514 | 0.9308 | 0.9297 |
107
+ | 0.218 | 6.05 | 780 | 0.2025 | 0.7870 | 0.8508 | 0.9364 | 0.9678 | 0.5859 | 0.8690 | 0.6870 | 0.9368 | 0.9360 | 0.9732 | 0.9282 | 0.4918 | 0.8198 | 0.6437 | 0.8651 | 0.8324 | 0.9282 | 0.9355 |
108
+ | 0.4344 | 6.2 | 800 | 0.2128 | 0.7816 | 0.8399 | 0.9361 | 0.9620 | 0.6012 | 0.8908 | 0.5972 | 0.9446 | 0.9092 | 0.9743 | 0.9243 | 0.5193 | 0.8213 | 0.5626 | 0.8684 | 0.8520 | 0.9233 | 0.9350 |
109
+ | 0.5841 | 6.36 | 820 | 0.2412 | 0.7965 | 0.8614 | 0.9378 | 0.9587 | 0.6847 | 0.8888 | 0.6398 | 0.9360 | 0.9408 | 0.9808 | 0.9254 | 0.5838 | 0.8259 | 0.6002 | 0.8758 | 0.8440 | 0.9204 | 0.9371 |
110
+ | 0.3048 | 6.51 | 840 | 0.2336 | 0.7869 | 0.8580 | 0.9344 | 0.9576 | 0.6990 | 0.8928 | 0.6175 | 0.9281 | 0.9381 | 0.9728 | 0.9160 | 0.5612 | 0.8190 | 0.5724 | 0.8720 | 0.8484 | 0.9193 | 0.9337 |
111
+ | 0.2002 | 6.67 | 860 | 0.2373 | 0.7929 | 0.8605 | 0.9343 | 0.9512 | 0.6492 | 0.8691 | 0.6968 | 0.9424 | 0.9290 | 0.9861 | 0.9216 | 0.5520 | 0.8087 | 0.6401 | 0.8685 | 0.8437 | 0.9155 | 0.9337 |
112
+ | 0.2093 | 6.82 | 880 | 0.2335 | 0.7918 | 0.8528 | 0.9364 | 0.9649 | 0.6226 | 0.8682 | 0.6653 | 0.9414 | 0.9311 | 0.9758 | 0.9210 | 0.5272 | 0.8166 | 0.6269 | 0.8720 | 0.8550 | 0.9235 | 0.9355 |
113
+ | 0.1581 | 6.98 | 900 | 0.2279 | 0.7995 | 0.8786 | 0.9368 | 0.9509 | 0.7124 | 0.8959 | 0.7476 | 0.9347 | 0.9273 | 0.9812 | 0.9159 | 0.5266 | 0.8244 | 0.6672 | 0.8800 | 0.8627 | 0.9198 | 0.9367 |
114
+ | 0.1209 | 7.13 | 920 | 0.2432 | 0.7832 | 0.8475 | 0.9327 | 0.9548 | 0.5902 | 0.8897 | 0.6726 | 0.9357 | 0.9173 | 0.9725 | 0.9104 | 0.5047 | 0.8096 | 0.6187 | 0.8725 | 0.8463 | 0.9201 | 0.9319 |
115
+ | 0.1492 | 7.29 | 940 | 0.2345 | 0.7920 | 0.8609 | 0.9360 | 0.9603 | 0.6503 | 0.8872 | 0.6902 | 0.9383 | 0.9340 | 0.9659 | 0.9194 | 0.5196 | 0.8231 | 0.6344 | 0.8728 | 0.8571 | 0.9179 | 0.9354 |
 
 
 
 
 
 
 
 
 
 
 
116
 
117
 
118
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0360467bed2ab0357e5f2f33d0aad7e707f66da8f765ff9b445676b7eee620aa
3
  size 256112508
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:852f1ec6e38e07b28af536e9153f983bb060fa0e108f31bc441f3e22bcc4806a
3
  size 256112508
runs/Mar06_13-03-38_4d04dfd18819/events.out.tfevents.1709730222.4d04dfd18819.27.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5c9df7c6381335033a5349289b2ef996db495169c2cae7eea7d52d0e1901a55
3
+ size 215746
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:04583144a48b81c7748118c27213c3c062210d08f04e08c863ab502e02e9da6d
3
  size 4792
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0101a4c64706838e0e9a33cc1577cc9dde33f175f08ac1423697be15f5dfd45
3
  size 4792