saketpatayeet
/

infection-segmentation

Image Segmentation

segmentation-models-pytorch

model_hub_mixin

pytorch_model_hub_mixin

semantic-segmentation

Model card Files Files and versions

saketpatayeet commited on 24 days ago

Commit

aa9fde7

·

verified ·

1 Parent(s): c82d091

Push model using huggingface_hub.

Files changed (3) hide show

README.md +3 -3
config.json +1 -1
model.safetensors +2 -2

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ model_init_params = {
     "decoder_attention_type": None,
     "decoder_interpolation": "nearest",
     "in_channels": 3,
-    "classes": 1,
     "activation": None,
     "aux_params": None
 }
@@ -47,8 +47,8 @@ model_init_params = {
 ```json
 [
     {
-        "test_per_image_iou": 0.7890859246253967,
-        "test_dataset_iou": 0.7133795022964478
     }
 ]
 ```

     "decoder_attention_type": None,
     "decoder_interpolation": "nearest",
     "in_channels": 3,
+    "classes": 3,
     "activation": None,
     "aux_params": None
 }
 ```json
 [
     {
+        "test_per_image_iou": 0.9501466155052185,
+        "test_dataset_iou": 0.9473266005516052
     }
 ]
 ```

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "_model_class": "UnetPlusPlus",
   "activation": null,
   "aux_params": null,
-  "classes": 1,
   "decoder_attention_type": null,
   "decoder_channels": [
     256,

   "_model_class": "UnetPlusPlus",
   "activation": null,
   "aux_params": null,
+  "classes": 3,
   "decoder_attention_type": null,
   "decoder_channels": [
     256,

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:576e1f0d1b1b5cbc2492fe4577551adc3ec0eb79cffc7a62f784ce8a23f51443
+size 63962676