Is there any way to generate a predicted txt file？

THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]

https://arxiv.org/abs/2405.14458

GNU Affero General Public License v3.0

9.58k stars 913 forks source link

Is there any way to generate a predicted txt file？ #290

Closed AZong76 closed 3 months ago

AZong76 commented 3 months ago

Similar to the following：

50,79,40,33,0.8213,4,-1,-1 98,41,39,27,0.7921,4,-1,-1

This is the code I predicted：

from ultralytics import YOLOv10 model = YOLOv10("/home/lyu4/datasets/dh_wp/dh/visdrone-det/yolov10-main/runs/detect/train4/weights/yolov10.pt") model.predict(source='/home/lyu4/datasets/dh_wp/dh/visdrone-det/visdrone2024/VisDrone2019-DET-test-challenge/images',save=True,show_labels=True)