HarDNet-MSEG: A Simple Encoder-Decoder Polyp Segmentation Neural Network that Achieves over 0.9 Mean Dice and 86 FPS

  • 2021-01-18 17:20:11
  • Chien-Hsiang Huang, Hung-Yu Wu, Youn-Long Lin
  • 0

Abstract

We propose a new convolution neural network called HarDNet-MSEG for polypsegmentation. It achieves SOTA in both accuracy and inference speed on fivepopular datasets. For Kvasir-SEG, HarDNet-MSEG delivers 0.904 mean Dice runningat 86.7 FPS on a GeForce RTX 2080 Ti GPU. It consists of a backbone and adecoder. The backbone is a low memory traffic CNN called HarDNet68, which hasbeen successfully applied to various CV tasks including image classification,object detection, multi-object tracking and semantic segmentation, etc. Thedecoder part is inspired by the Cascaded Partial Decoder, known for fast andaccurate salient object detection. We have evaluated HarDNet-MSEG using thosefive popular datasets. The code and all experiment details are available atGithub. https://github.com/james128333/HarDNet-MSEG

 

Quick Read (beta)

loading the full paper ...