|
--- |
|
license: mit |
|
--- |
|
|
|
## Model Card for butterfly_segmentation_unet |
|
|
|
This model takes in an image of a butterfly (with or without body attached to wings) and segments out any existing hindwings and forewings, in addition to pictured equipment described below. |
|
|
|
## Model Details |
|
|
|
unet_butterflies_256_256.hdf5 is the butterfly segmentation model. |
|
|
|
The segmentation model was trained on a dataset of 800 total images from the Jiggins, OM_STRI, and Monteiro datasets. The model architecture is based on a simple UNet architecture for multiclass classification. |
|
|
|
## Model Description |
|
|
|
Keras implementation of Butterfly UNet segmentation model. The model is responsible for taking an input image (256 x 256 x 3) and generating segmentation masks for all classes below that are found in the image. |
|
|
|
### Segmentation Classes |
|
- background |
|
- generic |
|
- right_forewing |
|
- left_forewing |
|
- right_hindwing |
|
- left_hindwing |
|
- ruler |
|
- white_balance |
|
- label |
|
- color_card |
|
- body |
|
|
|
**Developed by:** Michelle Ramirez |
|
|
|
## How to Get Started with the Model |
|
|
|
To view applications of how to load in the model file and predict masks on images, please refer to [this github repository](https://github.com/Imageomics/wing-segmentation) |
