1) Download the repository

git clone https://github.com/CEPrE-Unilurio/underwater_od.git

Before running any of the command below Make sure you are inside underwater_od/gym

2) Set up the environment

make setup

3) Download pre-trained models

make download_models
you should do this if and only if you do not have the pre-trained model already in you machine
this code will save the pre-trained model insideunderwater_od/ai4coral/pre-trained-models/

How to train a model

make model_name=ssd_mobilenet_v2_fpnlite train

How to evaluate a model

make model_name=ssd_mobilenet_v2_fpnlite eval
you should run this in parallel to model training

How to monitor model training

make model_name=ssd_mobilenet_v2_fpnlite monitor
you should run this in parallel to model training and model evaluation

How to export trained model

make model_name=ssd_mobilenet_v2_fpnlite export_to_tflite_graph

How to convert exported model to tflite

make model_name=ssd_mobilenet_v2_fpnlite convert_to_tflite

Note: model_name can be one the models listed in model underwater_od/gym/models.list

use make clean to clean up images and stoped containers, why?
well, you will need it when you are run out of space trust me I am an engineer ...