Awesome
CoralSeg: Learning coral segmentation from sparse annotations
Requirements
- tensorflow
- imgaug
- cv2
- keras
- scipy
- matplotlib
Citing CoralSeg
If you find our work useful in your research, please consider citing:
@inproceedings{alonso2019CoralSeg,
title={CoralSeg: Learning Coral Segmentation from Sparse Annotations},
author={Alonso, I{\~n}igo and Yuval, Matan and Eyal, Gal and Treibitz, Tali and Murillo, Ana C},
booktitle={Journal of Field Robotics},
year={2019}
}
Web project and Downloadable content (Weights, datasets...)
For downloading the trained weights and our datasets, please go to our website project
Multi-Level Superpixel Augmentation
The github repository concerning the labeling augmentation/propagation method is in here.
Training CoralSeg from Scratch
This repository already contains the EilatMixx dataset in the Datasets folder.
python train.py
This scripts has different arguments for specifying the dataset, the model path, the input resolution and other configuration parameters, feel free to play with them.
The training scripts gives test metrics in every epoch, but if you want to test it onced is trained, just change the train parameter to 0.
Training CoralSeg from CoralNet weights
First of all, download these weights and save them in the models folder. And the just execute:
python train_pretrained.py
In the script, the checkpoint_path parameter is set to models/deeplab_encoder that is where the trained weights should be.