Mask R-CNN with R-D Net: A Detection and Segmentation Model for Early Gastric Cancer Gastroscopy Images

Gastric cancer is one of the most common malignant tumors of the digestive system. Although gastroscopy can effectively detect early cancers, manual identification and diagnosis is time-consuming and laborious, and the false-negative rate is as high as 4.6-25.8%. Early gastric cancer lesion areas can be detected and segmented using deep learning techniques. We collected and organized 12665 eligible gastroscopy images, and cropped and expanded the original dataset after labeling by professional doctors, and finally obtained 19673 images for model training. We propose ResMSE Net and DWFF Net and apply these two networks to Mask R-CNN to obtain a new model: Mask R-CNN With R-D Net. On the same standard dataset, the mAP@[IoU=0.5] and ACC values of the Mask R-CNN With R-D Net model achieve respectively 45.37% and 67.74%, which are improved by 5.83% and 9.08%, respectively, compared with the original model. In our experiments, we found that Mask R-CNN With R-D Net is difficult to detect targets that are extremely similar to the background, which is the direction of our future work. Secondly, we hope to minimize the number of model parameters and make the model more lightweight under the premise of ensuring accuracy, so that the model can be embedded into gastroscopy detection equipment for real-time detection.