Push model using huggingface_hub.

README.md

@@ -3,9 +3,11 @@ library_name: segmentation-models-pytorch
 license: mit
 pipeline_tag: image-segmentation
 tags:
+- model_hub_mixin
+- pytorch_model_hub_mixin
+- segmentation-models-pytorch
 - semantic-segmentation
 - pytorch
-- segmentation-models-pytorch
 languages:
 - python
 ---
@@ -21,19 +23,20 @@ Table of Contents:
 ```python
 import segmentation_models_pytorch as smp
 
-model = smp.FPN.from_pretrained("calfin")
+model = smp.from_pretrained("<save-directory-or-this-repo>")
 ```
 
 ## Model init parameters
 ```python
 model_init_params = {
-    "encoder_name": "xception",
+    "encoder_name": "resnet34",
     "encoder_depth": 5,
     "encoder_weights": "imagenet",
     "decoder_pyramid_channels": 256,
     "decoder_segmentation_channels": 128,
     "decoder_merge_policy": "add",
     "decoder_dropout": 0.2,
+    "decoder_interpolation": "nearest",
     "in_channels": 3,
     "classes": 1,
     "activation": None,
@@ -46,8 +49,8 @@ model_init_params = {
 ```json
 [
     {
-        "test_per_image_iou": 0.8843663334846497,
-        "test_dataset_iou": 0.874630868434906
+        "test_per_image_iou": 0.5610896944999695,
+        "test_dataset_iou": 0.561133086681366
     }
 ]
 ```

config.json

@@ -4,11 +4,12 @@
   "aux_params": null,
   "classes": 1,
   "decoder_dropout": 0.2,
+  "decoder_interpolation": "nearest",
   "decoder_merge_policy": "add",
   "decoder_pyramid_channels": 256,
   "decoder_segmentation_channels": 128,
   "encoder_depth": 5,
-  "encoder_name": "xception",
+  "encoder_name": "resnet34",
   "encoder_weights": "imagenet",
   "in_channels": 3,
   "upsampling": 4

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cdc742e0fc9b65041f53ce49436c3329b36c190834bb1f6a2784b77510df38d6
-size 93212444
+oid sha256:e8e8290eeb068102ec28d1702b42ea13532451a15b2699460fd630b281623656
+size 92714324