HarDNet-MSEG: A Simple Encoder-Decoder Polyp Segmentation Neural Network that Achieves over 0.9 Mean Dice and 86 FPS
About
We propose a new convolution neural network called HarDNet-MSEG for polyp segmentation. It achieves SOTA in both accuracy and inference speed on five popular datasets. For Kvasir-SEG, HarDNet-MSEG delivers 0.904 mean Dice running at 86.7 FPS on a GeForce RTX 2080 Ti GPU. It consists of a backbone and a decoder. The backbone is a low memory traffic CNN called HarDNet68, which has been successfully applied to various CV tasks including image classification, object detection, multi-object tracking and semantic segmentation, etc. The decoder part is inspired by the Cascaded Partial Decoder, known for fast and accurate salient object detection. We have evaluated HarDNet-MSEG using those five popular datasets. The code and all experiment details are available at Github. https://github.com/james128333/HarDNet-MSEG
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Polyp Segmentation | CVC-ClinicDB (test) | DSC93.2 | 196 | |
| Polyp Segmentation | Kvasir | Dice Score91.2 | 128 | |
| Polyp Segmentation | ETIS | Dice Score70 | 108 | |
| Polyp Segmentation | Kvasir-SEG (test) | mIoU80.7 | 87 | |
| Polyp Segmentation | ETIS (test) | Mean Dice70 | 86 | |
| Polyp Segmentation | CVC-ClinicDB | Dice Coefficient90.9 | 81 | |
| Polyp Segmentation | ColonDB | mDice73.1 | 74 | |
| Polyp Segmentation | Kvasir (test) | Dice Coefficient91.2 | 73 | |
| Polyp Segmentation | CVC-ColonDB | mDice73.5 | 66 | |
| Polyp Segmentation | CVC-ColonDB (test) | Mean Dice0.731 | 62 |