Accepted At Ecological Informatics [Paper]
Download the annotated DeepFish dataset from here
To download OzFish dataset, visit the link
This projects was implemented using darknet framework for detection model. To know the setup requirements, how to compile and train on different OS, how to use command line everything you will get by visiting AlexeyAB/darknet github repository.
- Download the test dataset and unzip.
test dataset of deepfish
test dataset of ozfish
test dataset of merge dataset - Download the cfg file of a chosen model.
cfg of different models - Download the Model's weights.(google-drive mirror)
models trained on deepfish
models trained on ozfish
models trained on merge dataset - Content of the file obj.data should be
classes=1
train=data/train.txt
valid=data/test.txt
names=data/obj.names
backup=backup/
- obj.names files should contain just a 'Fish' word.
- Keep wieghts files in a backup directory and cfg files in cfg directory.
- Run the command(linux) to evaluate map for a YOLO-Fish models.
./darknet detector map data/obj.data cfg/yolo-fish-2.cfg backup/merge_yolo-fish-2.weights ./darknet detector demo data/obj.data cfg/yolo-fish-2.cfg backup/merge_yolo-fish-2.weights input.mp4 -dont_show -ext_output -out_filename output.avi