yolo v5 如何將辨識出的 label 提取為字串?

嘗試寫個簡單系統,想讓系統辨識完成後利用 if 判斷式做後續功能,請問如何將辨識的 label 做資料轉換成字串形式

img

img

简化model输出就是reg(float),conf(float),cls(int),其中你要的label是cls通过映射关系转换成str。
所以你可以使用cls(int)或label(str)存储都可以,后续的if判断懂点python就行

在detect.py里面作者已经写好了啊,

img


这里的label就是了,names[c]就是类名,conf为置信度。
或者你可以将其存为txt,在下面加上这个参数会将你的结果写成txt,不过里面是类别id+框位置

img