python - 我想用 YOLOv5 获取对象的特征值

Question

我想用 YOLOv5 获取对象的特征值。我猜在开源的“detect.py”中有一个提示。如何获取用于推理的对象的特征值？请告诉我如何解决。

Answer 1

在detect.py（第181行）中运行def里面的变量'det'，可以知道对象的xyxy值、置信度分数和类名的个数。

由于“det”是张量数据类型，因此您需要转换“det”。

如果您只想获取对象的类名的数量，您可以通过在 detect.py（第 205 行）中转换 cls 来轻松获取它，例如 'int(cls)'。