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

Question

我已按照yolov5中的教程进行操作。当我完成火车并开始测试时，我想根据标签计算有多少对象。在这里，我将此代码添加到文件 detect.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