Just like Mask R-CNN is an extension of Faster R-CNN for instance segmentation, I was looking for a similar extension for YOLO, but couldn't find any. Solutions in pytorch would be especially useful.
There are YOLOv3 modification which is more precise and accurate, and easily can be trained, search the github for the keras version of yolov3 it's one of the best versions of yolov3 implementation https://github.com/qqwweee/keras-yolo3