Как найти значения пикселей объектов, обнаруженных из yolo в python?

Question

Я использую yolo для обнаружения изображений в моем настраиваемом наборе данных, где я тренируюсь, чтобы идентифицировать определенный объект в наборе данных. А после обучения алгоритм правильно предсказал эти объекты на изображениях. Я хотел бы получить значения rbg только для предсказанных объектов, в opencv я не нашел способа проверить значения rgb для объектов внутри обнаруженного поля. На прилагаемом образце изображения я хотел бы видеть значение rgb обнаруженной чашки.

Answer 1

1. Получить координату ограничительной рамки объекта

Например, result = [x, y, w, h] или [x1, y1, x2, y2]

Проверьте изображение, чтобы получить значение RGB, используя координату bbox

Например, если ваш результат [x1, y1, x2, y2] и ваша форма изображения img [640,480,3]

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

вы получите значение rgb обнаруженного вами объекта