README.md

metadata

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:

from transformers import AutoModelForObjectDetection, AutoFeatureExtractor

model = AutoModelForObjectDetection.from_pretrained("Dreamy0/GermiNet-instance-segmentation")
feature_extractor = AutoFeatureExtractor.from_pretrained("Dreamy0/GermiNet-instance-segmentation")