How to get the bound box coordinates of a prediction?

hukaixuan19970627 / yolov5_obb

yolov5 + csl_label.(Oriented Object Detection)（Rotation Detection）（Rotated BBox）基于yolov5的旋转目标检测

GNU General Public License v3.0

1.81k stars 425 forks source link

How to get the bound box coordinates of a prediction? #491

Open OBaruch opened 1 year ago

OBaruch commented 1 year ago

After executing an inference whit:

python detect.py --weights 'runs/train/yolov5m_csl_dotav1.5/weights/best.pt' --source 'dataset/dataset_demo/images/' --img 2048 --device 0 --conf-thres 0.25 --iou-thres 0.2 --hide-labels --hide-conf

how to get the bound box coordinates of the predictions?

Wangfeng2394 commented 1 year ago

输出的参数是向量，你需要解析

ibtehajali67 commented 1 year ago

just add --save_txt, it save a txt file for every image . so in this file from 1 to 9 these are boundry box value in every line of text file