This repository created to train a model (Pre-trained Faster R-CNN ResNet-50 FPN Model)and detect an underwater object using pytorch API with the help of Detecto.
See works with other frameworks: tensorflow, openCV.
1. Install Detecto
pip3 install detecto
3. Use labelimg to label images.
pip3 install labelImg
from detecto.utils import xml_to_csv
xml_to_csv('path_to_xml_folder/', 'labels.csv')
- Train on your PC, run this script.
- Train on a powerful machine, run colab notebook on Google Colab.
- Test using openCV, run this script.