WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

How to get the bounding box coordinates of Yolor output ? #261

Open HissaanAli opened 2 years ago

HissaanAli commented 2 years ago

Running the detect.py I can get the image with bounding box drawn around the objects, but I can't figure out how to get just the coordinates of the bounding boxes of objects detected ? Yolov5 provides those coordinates very easily, but I couldn't find anything similar in YoloR. Do you have any idea how would i get those coordinates ? Any help would be appreciated.