End of training

README.md

@@ -2,6 +2,8 @@
 license: mit
 base_model: openmmlab/upernet-swin-small
 tags:
+- image-segmentation
+- vision
 - generated_from_trainer
 model-index:
 - name: upernet-swin-small-finetuned
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # upernet-swin-small-finetuned
 
-This model is a fine-tuned version of [openmmlab/upernet-swin-small](https://huggingface.co/openmmlab/upernet-swin-small) on an unknown dataset.
+This model is a fine-tuned version of [openmmlab/upernet-swin-small](https://huggingface.co/openmmlab/upernet-swin-small) on the jpodivin/plantorgans dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.2914
 - Mean Iou: 0.4182

all_results.json

@@ -0,0 +1,25 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy_Flower": 0.0,
+    "eval_accuracy_Fruit": 0.8589832879841797,
+    "eval_accuracy_Leaf": 0.7032167577488816,
+    "eval_accuracy_Stem": 0.5504917857429176,
+    "eval_accuracy_void": NaN,
+    "eval_iou_Flower": 0.0,
+    "eval_iou_Fruit": 0.8553766723933364,
+    "eval_iou_Leaf": 0.697605663193013,
+    "eval_iou_Stem": 0.5380745361554214,
+    "eval_iou_void": 0.0,
+    "eval_loss": 0.2914386987686157,
+    "eval_mean_accuracy": 0.5281729578689947,
+    "eval_mean_iou": 0.41821137434835415,
+    "eval_median_iou": 0.5380745530128479,
+    "eval_overall_accuracy": 0.7341365933972603,
+    "eval_runtime": 722.6168,
+    "eval_samples_per_second": 1.989,
+    "eval_steps_per_second": 0.199,
+    "train_loss": 0.4155007145629413,
+    "train_runtime": 4721.6055,
+    "train_samples_per_second": 3.65,
+    "train_steps_per_second": 0.365
+}

eval_results.json

@@ -0,0 +1,21 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy_Flower": 0.0,
+    "eval_accuracy_Fruit": 0.8589832879841797,
+    "eval_accuracy_Leaf": 0.7032167577488816,
+    "eval_accuracy_Stem": 0.5504917857429176,
+    "eval_accuracy_void": NaN,
+    "eval_iou_Flower": 0.0,
+    "eval_iou_Fruit": 0.8553766723933364,
+    "eval_iou_Leaf": 0.697605663193013,
+    "eval_iou_Stem": 0.5380745361554214,
+    "eval_iou_void": 0.0,
+    "eval_loss": 0.2914386987686157,
+    "eval_mean_accuracy": 0.5281729578689947,
+    "eval_mean_iou": 0.41821137434835415,
+    "eval_median_iou": 0.5380745530128479,
+    "eval_overall_accuracy": 0.7341365933972603,
+    "eval_runtime": 722.6168,
+    "eval_samples_per_second": 1.989,
+    "eval_steps_per_second": 0.199
+}

train_results.json

@@ -0,0 +1,7 @@
+{
+    "epoch": 3.0,
+    "train_loss": 0.4155007145629413,
+    "train_runtime": 4721.6055,
+    "train_samples_per_second": 3.65,
+    "train_steps_per_second": 0.365
+}