除了在 yolov5 上使用 pandas 之外，你还能得到返回的结果吗？

Question

在 yolov5 教程中，它说 pandas().xyxy[0] 从推理中获得结果。

results = model(testa) # inference

boxes = results.pandas().xyxy[0] # gets the results

但是是否可以使用熊猫以外的其他东西（如极地）来获得返回的结果？因为使用 pandas 太慢了，我的代码需要更快。 如果有人能告诉我，我将不胜感激（一个简单的例子也很好！）

Answer 1

您可以使用 NumPy 处理检测。不清楚你想要什么。就我而言，基于检测类和 X0 坐标，我想做出决定。 “boxes”数组分别由 bbox 坐标、置信度分数和类 id 组成。我使用了返回标准化坐标的“xywhn”格式。所以我检查每一行的最后一个元素和第一个元素，“wanted”数组返回满足条件的行索引。

results = model(img)
boxes = results.xywhn[0].detach().cpu().numpy()
wanted = np.where(((boxes[:,-1] == 1) & (boxes[:,0] > 0.4)))

print(results)
print(wanted)

[[    0.64899     0.64092    0.096527     0.25932     0.92479           2]
 [    0.72146     0.70188    0.058451     0.14613     0.90287           1]
 [    0.71426       0.931    0.091868     0.11923     0.88958           3]]
(array([1]),)

如果您不需要存储检测并立即处理它们，此解决方案可能会起作用。