Kvasir-Instruments and Polyp Segmentation Using UNet
Original version
Pandey, S., Keprate, A. (2021) Kvasir-Instruments and Polyp Segmentation Using UNet. Nordic Machine Intelligence (NMI), 1(1) 10.5617/nmi.9130Abstract
This paper aims to describe the methodology used to develop, fine-tune and analyze a UNet model for creating masks for two datasets: Polyp Segmentation and Instrument Segmentation, which are part of MedAI challenge. For training and validation, we have used the same methodology on both tasks and finally on the hidden testing dataset the model resulted in an accuracy of 0.9721, dice score of 0.7980 for the instrumentation task, and the accuracy of 0.5646 and a dice score of 0.4100 was achieved for the Polyp Segmentation.