Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

Chien-Hsiang Huang, Hung-Yu Wu, Youn-Long Lin• 2021

Related benchmarks

TaskDatasetResultRank
Polyp SegmentationCVC-ClinicDB (test)
DSC93.2
196
Polyp SegmentationKvasir
Dice Score91.2
128
Polyp SegmentationETIS
Dice Score70
108
Polyp SegmentationKvasir-SEG (test)
mIoU80.7
87
Polyp SegmentationETIS (test)
Mean Dice70
86
Polyp SegmentationCVC-ClinicDB
Dice Coefficient90.9
81
Polyp SegmentationColonDB
mDice73.1
74
Polyp SegmentationKvasir (test)
Dice Coefficient91.2
73
Polyp SegmentationCVC-ColonDB
mDice73.5
66
Polyp SegmentationCVC-ColonDB (test)
Mean Dice0.731
62
Showing 10 of 52 rows

Other info

Code

Follow for update