【发布时间】：2021-09-10 10:18:21
【问题描述】：
我想用 YOLOv5 获取对象的特征值。我猜在开源的“detect.py”中有一个提示。 如何获取用于推理的对象的特征值？请告诉我如何解决。
【问题讨论】：
标签： python pytorch image-recognition yolov5
【发布时间】：2021-09-10 10:18:21
【问题描述】：
在detect.py(第181行)中运行def里面的变量'det'，可以知道对象的xyxy值、置信度分数、类名个数。
由于 'det' 是张量数据类型，您需要转换 'det'。
如果只想获取对象的类名个数，可以通过在detect.py(line 205)中像'int(cls)'那样转换cls来轻松获取。
【讨论】：