如何找到在python中从yolo检测到的对象的像素值?

问题描述 投票:0回答:1

我正在使用yolo对自定义数据集进行图像检测,在这里我训练以识别数据集中的某些对象。训练后,该算法可正确预测图像中的这些对象。我想单独获得预测对象的rbg值,在opencv中,我没有找到一种方法来检查检测到的框内对象的rgb值。在所附的示例图像中,我想查看检测到的杯子的rgb值。

enter image description here

python opencv object-detection color-detection
1个回答
0
投票
  1. 获取对象的边界框坐标

例如,结果= [x,y,w,h]或[x1,y1,x2,y2]

  1. 使用bbox坐标检查图像以获取RGB值

例如,如果您的结果为[x1,y1,x2,y2],并且图像形状为img [640,480,3]

rgb_values = img[x1:x2, y1:y2, 0:2]

您将获得检测到的对象的rgb值

© www.soinside.com 2019 - 2024. All rights reserved.