使用坐标和 ID 等 deepSORT 预测信息保存 YOLOv4

Question

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

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

如有任何帮助，我们将不胜感激！

Answer 1

您可以修改 object_tracker.py、line 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

使用坐标和 ID 等 deepSORT 预测信息保存 YOLOv4

Save YOLOv4 with deepSORT prediction info such as coords and ID

python

yolo