A new data label conversion algorithm for YOLO segmentation of medical images

Muhammed Telceken,Mehmet Okuyar,Devrim Akgun,Sezgin Kacar,Mehmet Sait Vural
DOI: https://doi.org/10.1140/epjs/s11734-024-01338-5
2024-10-03
The European Physical Journal Special Topics
Abstract:Object detection and segmentation architectures like YOLO have increased accuracy and speed, making them more practical in medical applications, including polyp identification for medical imaging and diagnosis. However, accurately defining segmentation boundaries to train a YOLO model for medical diagnosis requires considerable manual effort. This study developed a new method for automatically generating labels from ground-truth masks to process datasets for training YOLO-based segmentation algorithms. The developed method offers a simple interface for users to readily sample segmentation masks in datasets into labels compatible with the YOLO format. The technique involves performing contour extraction while processing labeled masks to ensure that the resulting contour information represents the object boundaries and makes it possible to map them according to the YOLO format. The efficiency of the method has been shown via training YOLOv5, YOLOv7, and YOLOv8-based models. Comparative results on the polyp segmentation dataset show successful improvements using YOLO-based segmentation over the literature.
physics, multidisciplinary
What problem does this paper attempt to address?