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

Abstract

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

References in corpus (2)

• Real-Time Polyp Detection, Localization and Segmentation in Colonoscopy Using Deep Learning
• Automated polyp detection in colon capsule endoscopy

Cited by in corpus (6)

• FCN-Transformer Feature Fusion for Polyp Segmentation
• TransFuse: Fusing Transformers and CNNs for Medical Image Segmentation
• Edge-aware Feature Aggregation Network for Polyp Segmentation
• Diabetic Foot Ulcer Grand Challenge 2022 Summary
• Joint one-sided synthetic unpaired image translation and segmentation for colorectal cancer prevention
• An Enhanced Harmonic Densely Connected Hybrid Transformer Network Architecture for Chronic Wound Segmentation Utilising Multi-Colour Space Tensor Merging