- Python3
- PyTorch >= 0.4.1
- OpenCV
- Imgaug (https://github.com/aleju/imgaug)
- Pillow-SIMD (https://github.com/uploadcare/pillow-simd)
https://drive.google.com/drive/folders/1Fh1mTowJR1KuVhCTRAnEL28UpKUyD7lx?usp=sharing (1)
Training (Network configuration is defined in config-raccoon.py)
python3 train.py config-raccoon.py
Predict (Download racccoon/best.pth from above link and put in trainingLogs/raccoonDataset/best.pth)
python3 detect.py config-raccoon.py datasets/raccoon/raccoon-10.jpg
- Darknet53
- Darknet21 (WIP)
- NASNetAMobile
- Training visualization using TensorboardX
- MAP evaluation
- User friendly documents
- More backbone networks
- CPU only Inference