Как создать бинарную маску из результата сегментации YOLO8 ⇐ Python

[Anonymous]

Я хочу сегментировать изображение с использованием yolo8, а затем создать маску для всех объектов на изображении с конкретным классом. < /p>
Я разработал этот код: < /p>

Код: Выделить всё

img=cv2.imread('images/bus.jpg')
model = YOLO('yolov8m-seg.pt')
results = model.predict(source=img.copy(), save=False, save_txt=False)
class_ids = np.array(results[0].boxes.cls.cpu(), dtype="int")
for i in range(len(class_ids)):
if class_ids[i]==0:
empty_image = np.zeros((height, width,3), dtype=np.uint8)
res_plotted = results[0][i].plot(boxes=0, img=empty_image)

В приведенном выше коде res_plotted является маской для одного объекта, в RGB. Я хочу добавить все эти изображения друг другу и создать маску для всех объектов с классом 0 (в этом примере это пешеход)
Мои вопросы:

• Как я могу завершить этот код? > Есть ли в библиотеке YOLO8 какая -то функция утилиты?

Подробнее здесь: https://stackoverflow.com/questions/761 ... ion-result