Upload 5 files

README.md

@@ -1,3 +1,17 @@
----
-license: mit
----
+# Instance Segmentation Model
+
+## Description
+This model performs instance segmentation using Mask R-CNN. It was trained on a custom dataset containing [X] images with [Y] classes.
+
+## Training Data
+- Dataset: [Germination Images]
+- Number of Images: [22]
+- Number of Classes: [2]
+
+## Usage
+Load the model using:
+```python
+from transformers import AutoModelForObjectDetection, AutoFeatureExtractor
+
+model = AutoModelForObjectDetection.from_pretrained("Dreamy0/GermiNet-instance-segmentation")
+feature_extractor = AutoFeatureExtractor.from_pretrained("Dreamy0/GermiNet-instance-segmentation")

config.json

@@ -0,0 +1,10 @@
+{
+    "model_type": "detectron2",
+    "architectures": ["MaskRCNN"],
+    "num_classes": 3, 
+    "input_size": [800, 1333], 
+    "pretrained": false,
+    "max_iter": 1000,
+    "batch_size": 2,
+    "learning_rate": 0.00025
+  }

label_map.json

@@ -0,0 +1,5 @@
+{
+    "0": "background",
+    "1": "Normal",
+    "2": "Abnormal"
+  }

preprocessor_config.json

@@ -0,0 +1,6 @@
+{
+    "image_mean": [0.485, 0.456, 0.406],
+    "image_std": [0.229, 0.224, 0.225],
+    "size_divisibility": 32,
+    "input_format": "RGB"
+  }

pytorch_model.bin

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