Я обучил модель детектора 2 с помощью одного класса. Теперь я хочу установить белый цвет для всего, что не находится внутри bbox, а все остальное оставить как есть. На одном изображении может быть несколько bbox, и они могут накладываться друг на друга.
Я прочитал документацию по детектору2, а также по cv2, но не смог найти решение своей проблемы.
Это мой код для прогнозов:
from detectron2 import model_zoo
from detectron2.config import get_cfg
from detectron2.engine import DefaultPredictor
import cv2
cfg = get_cfg()
cfg.merge_from_file(model_zoo.get_config_file('COCO-Detection/tmp'))
cfg.MODEL.WEIGHTS = 'tmp'
predictor = DefaultPredictor(cfg)
img = cv2.imread('tmp')
out = predictor(img)
Подробнее здесь: https://stackoverflow.com/questions/786 ... detectron2
