End of training

README.md

@@ -1,10 +1,10 @@
 ---
 license: other
+base_model: nvidia/mit-b0
 tags:
 - vision
 - image-segmentation
 - generated_from_trainer
-base_model: nvidia/mit-b0
 model-index:
 - name: segformer-b0-finetuned-agriculture
   results: []
@@ -17,35 +17,22 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the None dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.3715
-- eval_mean_iou: 0.3376
-- eval_mean_accuracy: 0.4059
-- eval_overall_accuracy: 0.6146
-- eval_accuracy_unlabeled: nan
-- eval_accuracy_double_plant: nan
-- eval_accuracy_drydown: nan
-- eval_accuracy_endrow: nan
-- eval_accuracy_nutrient_deficiency: 0.6788
-- eval_accuracy_planter_skip: 0.0031
-- eval_accuracy_storm_damage: nan
-- eval_accuracy_water: 0.7197
-- eval_accuracy_waterway: 0.0120
-- eval_accuracy_weed_cluster: 0.6158
-- eval_iou_unlabeled: 0.0
-- eval_iou_double_plant: nan
-- eval_iou_drydown: nan
-- eval_iou_endrow: nan
-- eval_iou_nutrient_deficiency: 0.6772
-- eval_iou_planter_skip: 0.0031
-- eval_iou_storm_damage: nan
-- eval_iou_water: 0.7195
-- eval_iou_waterway: 0.0119
-- eval_iou_weed_cluster: 0.6137
-- eval_runtime: 241.6207
-- eval_samples_per_second: 7.495
-- eval_steps_per_second: 3.75
-- epoch: 0.74
-- step: 6000
+- Loss: 0.3305
+- Mean Iou: 0.4242
+- Mean Accuracy: 0.5107
+- Overall Accuracy: 0.6733
+- Accuracy Unlabeled: nan
+- Accuracy Nutrient Deficiency: 0.6872
+- Accuracy Planter Skip: 0.1915
+- Accuracy Water: 0.8549
+- Accuracy Waterway: 0.1797
+- Accuracy Weed Cluster: 0.6404
+- Iou Unlabeled: 0.0
+- Iou Nutrient Deficiency: 0.6865
+- Iou Planter Skip: 0.1914
+- Iou Water: 0.8475
+- Iou Waterway: 0.1795
+- Iou Weed Cluster: 0.6401
 
 ## Model description
 
@@ -64,7 +51,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0001
+- learning_rate: 1e-05
 - train_batch_size: 2
 - eval_batch_size: 2
 - seed: 42
@@ -72,6 +59,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 5
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Unlabeled | Accuracy Nutrient Deficiency | Accuracy Planter Skip | Accuracy Water | Accuracy Waterway | Accuracy Weed Cluster | Iou Unlabeled | Iou Nutrient Deficiency | Iou Planter Skip | Iou Water | Iou Waterway | Iou Weed Cluster |
+|:-------------:|:-----:|:-----:|:---------------:|:--------:|:-------------:|:----------------:|:------------------:|:----------------------------:|:---------------------:|:--------------:|:-----------------:|:---------------------:|:-------------:|:-----------------------:|:----------------:|:---------:|:------------:|:----------------:|
+| 0.2889        | 1.0   | 8145  | 0.4127          | 0.2578   | 0.3110        | 0.4484           | nan                | 0.3062                       | 0.0                   | 0.7988         | 0.0007            | 0.4496                | 0.0           | 0.3062                  | 0.0              | 0.7913    | 0.0007       | 0.4485           |
+| 0.3157        | 2.0   | 16290 | 0.3877          | 0.3374   | 0.4070        | 0.5970           | nan                | 0.5241                       | 0.0023                | 0.8816         | 0.0303            | 0.5969                | 0.0           | 0.5237                  | 0.0023           | 0.8715    | 0.0301       | 0.5968           |
+| 0.2637        | 3.0   | 24435 | 0.3531          | 0.3717   | 0.4480        | 0.6171           | nan                | 0.5638                       | 0.0409                | 0.8804         | 0.1563            | 0.5984                | 0.0           | 0.5633                  | 0.0409           | 0.8723    | 0.1554       | 0.5982           |
+| 0.4715        | 4.0   | 32580 | 0.3337          | 0.3653   | 0.4398        | 0.6073           | nan                | 0.6172                       | 0.1068                | 0.8077         | 0.0976            | 0.5698                | 0.0           | 0.6164                  | 0.1068           | 0.8015    | 0.0976       | 0.5696           |
+| 0.0668        | 5.0   | 40725 | 0.3305          | 0.4242   | 0.5107        | 0.6733           | nan                | 0.6872                       | 0.1915                | 0.8549         | 0.1797            | 0.6404                | 0.0           | 0.6865                  | 0.1914           | 0.8475    | 0.1795       | 0.6401           |
+
+
 ### Framework versions
 
 - Transformers 4.39.1

config.json

@@ -29,29 +29,21 @@
   ],
   "id2label": {
     "0": "unlabeled",
-    "1": "double_plant",
-    "2": "drydown",
-    "3": "endrow",
-    "4": "nutrient_deficiency",
-    "5": "planter_skip",
-    "6": "storm_damage",
-    "7": "water",
-    "8": "waterway",
-    "9": "weed_cluster"
+    "1": "nutrient_deficiency",
+    "2": "planter_skip",
+    "3": "water",
+    "4": "waterway",
+    "5": "weed_cluster"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "double_plant": 1,
-    "drydown": 2,
-    "endrow": 3,
-    "nutrient_deficiency": 4,
-    "planter_skip": 5,
-    "storm_damage": 6,
+    "nutrient_deficiency": 1,
+    "planter_skip": 2,
     "unlabeled": 0,
-    "water": 7,
-    "waterway": 8,
-    "weed_cluster": 9
+    "water": 3,
+    "waterway": 4,
+    "weed_cluster": 5
   },
   "layer_norm_eps": 1e-06,
   "mlp_ratios": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ebfda59c211aee7ded3da739317d67635b4594f87b107f700642556dc4280a0a
-size 14893008
+oid sha256:eb96e9f89acda4776d613a0e3b5a5a48b193fbb5eaa8bade3700e49817564900
+size 14888896

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f6773f924ac0d79331eec0fd8610773c19ead832e76f01537a51afe67bb3cee1
+oid sha256:cb3bdebfa298c457c4727d0d8ae4ada13e21ef568ce78578c36298b37e8b161e
 size 4984