README.md

---
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)