File size: 823 Bytes
169acbb 59ba540 169acbb f39a6b8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
---
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") |