How to do semantic segmentation using yolov2 in python3.7?

Question

I labeled images with bounding box and then used yolov2 (python 3.7) for object detection and It was successful (https://github.com/deep-diver/Soccer-Ball-Detection-YOLOv2/blob/master/YOLOv2-Train.ipynb). I need to use yolov2 for semantic segmentation using python. Which image-labeling software can I use for semantic segmentation which its output is compatible with yolo? How can I modify the code in the aforementioned website to use yolov2 for this matter? Or do we have any example for that?

Thank you for your help in advance.

Abbas,

Answer 1

None of the YOLO algorithms perform Semantic Segmentation. They are Object Detection algorithms. In your case, you can use any of the following algorithms for segmentation: DeepLab, UNet, PSPNet, EfficientNet, etc.

Answer 2

Hello You can use YOLACT/YOLACT++ https://github.com/dbolya/yolact It uses same labelme annotations and you just have to edit the cofig.py a little and you will get the desired results. I am using it and it is superb.

Yolact is a kind of transformation of YOLO and it means You Look at the Coefficient.