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

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

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

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

  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值

本文链接:https://www.f2er.com/3145000.html

大家都在问