Skip to content

Semantic Segmentation of Underwater Imagery: Dataset and Benchmark. #IROS2020

License

Notifications You must be signed in to change notification settings

xahidbuffon/SUIM

Repository files navigation

Repository for the paper Semantic Segmentation of Underwater Imagery: Dataset and Benchmark (IROS 2020) det-data

SUIM Dataset

  • For semantic segmentation of natural underwater images
  • 1525 annotated images for training/validation and 110 samples for testing
  • BW: Background/waterbody • HD: human divers • PF: Aquatic plants and sea-grass • WR: Wrecks/ruins
  • RO: Robots/instruments • RI: Reefs/invertebrates • FV: Fish and vertebrates • SR: Sea-floor/rocks

SUIM-Net Model

  • A fully-convolutional encoder-decoder network
  • SUIM-Net (RSB): simple and light model; offers reasonable performance at a fast rate
  • SUIM-Net (VGG): provides better generalization performance
  • Detailed architecture is in models; associated train/test scripts are also provided
  • The get_f1_iou.py script is used for performance evaluation

Benchmark Evaluation

  • Performance analysis for semantic segmentation and saliency prediction
  • SOTA models in comparison: • FCNUNetSegNetPSPNetDeepLab-v3
  • Metrics: • region similarity (F score) and • contour accuracy (mIOU)
  • Download experimental data from here and checkpoints data from here

det-data det-data

Bibliography Entry

@inproceedings{islam2020suim,
  title={{Semantic Segmentation of Underwater Imagery: Dataset and Benchmark}},
  author={Islam, Md Jahidul and Edge, Chelsey and Xiao, Yuyang and Luo, Peigen and Mehtaz, 
              Muntaqim and Morse, Christopher and Enan, Sadman Sakib and Sattar, Junaed},
  booktitle={IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS)},
  year={2020},
  organization={IEEE/RSJ}
}

Acknowledgements