问使用YOLOv4预测信息(如同弦和ID )保存deepSORT
EN

Stack Overflow用户

提问于 2022-04-13 20:11:32

回答 1查看 222关注 0票数 -1

我正在学习本教程(https://github.com/theAIGuysCode/yolov4-deepsort)。现在，我希望根据类、ID和坐标位置将预测保存到一个文件中(csv/txt/ and )。我在网上查过了，但似乎找不到解决办法。我特别需要ID值。

我可以使用参数-信息在Anaconda提示符中显示这个信息，但是我只需要保存文件中的信息。

任何帮助都将不胜感激！

yolo

python

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-04-13 21:09:45

您可以修改object_tracker.py、第219项，并将信息保存到文件中：

with open('output.txt', 'a') as f:
    f.write("Tracker ID: {}, Class: {},  BBox Coords (xmin, ymin, xmax, ymax): {}".format(str(track.track_id), class_name, (int(bbox[0]), int(bbox[1]), int(bbox[2]), int(bbox[3])))))
f.close

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/71863346

复制

相似问题

问使用YOLOv4预测信息(如同弦和ID )保存deepSORT
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问使用YOLOv4预测信息(如同弦和ID )保存deepSORTEN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问使用YOLOv4预测信息(如同弦和ID )保存deepSORT
EN