|
--- |
|
language: en |
|
license: mit |
|
tags: |
|
- instance-segmentation |
|
- detectron2 |
|
- pytorch |
|
datasets: |
|
- custom-dataset |
|
metrics: |
|
- mean-average-precision |
|
pipeline_tag: object-detection |
|
labels: |
|
- background |
|
- germinated |
|
- non-germinated |
|
--- |
|
|
|
|
|
# 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") |