extract_bounding_boxes
该脚本将帮助您提取文件ID和YOLOV3 / 4/5中检测到的对象的相应边界框。 一旦训练并验证了YOLOV3 / 4/5模型,便会创建一个predicted.json文件,其中包含“文件ID”和所有验证图像集中所有检测到的所选对象的相对边界框坐标。 该脚本将帮助您根据设置的置信度阈值提取所有边界框坐标和相应的文件ID,并将其写入新的csv文件中,此外,它还可以统计检测到的对象总数并将其打印出来。
2021-02-16 12:18:46
1KB
Python
1