Butterfly Detection

This is a two-stage object detection model for detecting butterflies in images.

Usage

pip install -r requirements.txt

Place the dataset in the following structure:

datasets
├─ Annotations
│  ├─ IMG_000001.xml
│  └─ ...
├─ JPEGImages
│  ├─ IMG_000001.jpg
│  └─ ...
└─ TestData
   ├─ IMG_000001.jpg
   └─ ...

Run train_detection.ipynb.
You can test the model with detection.py.

Run train.ipynb.
You can test the model with classification.py.

Name		Name	Last commit message	Last commit date
Latest commit History 25 Commits
.gitignore		.gitignore
README.md		README.md
annotation_parser.py		annotation_parser.py
categories.json		categories.json
classification.py		classification.py
dataset_models.py		dataset_models.py
detection.py		detection.py
predict.py		predict.py
preprocess.py		preprocess.py
requirements.txt		requirements.txt
selective_search.py		selective_search.py
train.ipynb		train.ipynb
train_detection.ipynb		train_detection.ipynb
util.py		util.py