任何人都可以帮我如何使用yolov5计算对象吗？

Question

我已经按照yolov5中的教程进行操作。当我完成训练并开始测试时，我想根据标签计算有多少个对象。 在这里，我在文件 detector.py 中添加了这段代码

class_name_count = 'tree'
            l = s[1:s.find(class_name_count)].split()[-1]
            if class_name_count in s:
                print(l,class_name_count)
                cv2.rectangle(im0, (0,0), (1100, 250), -1)
                cv2.putText(im0,1 + class_name_count,(0,200), cv2.FONT_HERSHEY_SIMPLEX, 2,(255,255,255),8,cv2.LINE_AA)

但是输出是：

类型错误：+ 不支持的操作数类型：“int”和“str”

有什么解决办法吗？

Answer 1

您可以将结果保存到 *.txt 文件，然后导入。只需添加这一行：--save_txt

示例：

python detect.py --source input/image_1.jpg --weights yolov5x.pt --project infer_yolov5x --save-txt

Answer 2

你基本上无法将字符串和整型变量相加。 请参考此链接：https://github.com/ultralytics/yolov5/issues/2696