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

Question

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

标签： pythonopencvobject-detectioncolor-detection

Answer 1

1. 获取对象的边界框坐标

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

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

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

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

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