Upload trained UnetPlusPlus model

README.md

@@ -1,3 +1,30 @@
----
-license: mit
----
+---
+license: mit
+tags:
+- computer-vision
+- image-segmentation
+- pytorch
+- efficientnet
+- plant-disease
+- segmentation-models-pytorch
+---
+# nwrd-tukl
+UnetPlusPlus model with timm-efficientnet-b7 encoder for plant disease segmentation.
+## Model Details
+- **Architecture:** UnetPlusPlus
+- **Encoder:** timm-efficientnet-b7
+- **Dataset:** [Plant Disease Segmentation](https://www.kaggle.com/datasets/vipoooool/plantdiseasestukl)
+- **Image Size:** 512x512
+- **Classes:** 2 (0: Background, 1: Disease)
+## How to Use
+```bash
+pip install torch segmentation-models-pytorch huggingface_hub
+```
+```python
+import torch
+import segmentation_models_pytorch as smp
+from huggingface_hub import hf_hub_download
+model = smp.UnetPlusPlus(encoder_name="timm-efficientnet-b7", encoder_weights=None, in_channels=3, classes=2)
+model.load_state_dict(torch.load(hf_hub_download(repo_id="KM-Alee/nwrd-tukl", filename="pytorch_model.bin")))
+model.eval()
+```

config.json

@@ -0,0 +1,15 @@
+{
+    "architecture": "segmentation_models_pytorch.UnetPlusPlus",
+    "model_name": "UnetPlusPlus",
+    "encoder": "timm-efficientnet-b7",
+    "patch_size": 512,
+    "num_classes": 2,
+    "class_values": [
+        0,
+        255
+    ],
+    "class_mapping": {
+        "0": "Background",
+        "1": "Disease"
+    }
+}

pytorch_model.bin

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