DeepFish (A Realistic Fish-Habitat Dataset to Evaluate Algorithms for Underwater Visual Analysis) [Paper]
This repository contains the code to reproduce the experiments of the paper.
pip install -r requirements.txt
Download the repository:
git clone https://github.com/alzayats/DeepFish.git
Experiment hyperparameters are defined in exp_configs.py
Run the following command to reproduce the experiments in the paper:
python trainval.py -e ${TASK} -sb ${SAVEDIR_BASE} -d ${DATADIR} -r 1
The variables (${...}
) can be substituted with the following values:
TASK
: clf, reg, loc, segSAVEDIR_BASE
: Absolute path to where results will be savedDATADIR
: Absolute path containing the downloaded datasets
If you use the DeepFish dataset in your work, please cite it as:
@article{saleh2020realistic,
title={A Realistic Fish-Habitat Dataset to Evaluate Algorithms for Underwater Visual Analysis},
author={Alzayat Saleh and Issam H. Laradji and Dmitry A. Konovalov and Michael Bradley and David Vazquez and Marcus Sheaves},
year={2020},
eprint={2008.12603},
archivePrefix={arXiv},
primaryClass={cs.CV}
}