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

Question

Я использовал последовательную модель, включающую несколько сверточных слоев, чтобы отличить большой палец от указательного на изображениях. Обученная модель довольно хорошо распознает, есть ли на картинке большой или указательный палец. Теперь я хочу добавить скрипт, чтобы нарисовать рамку вокруг распознанного пальца на новом изображении, на которое я хочу применить модель. Мне нужна коробка для извлечения пальца из изображения после этапа распознавания. может кто-нибудь мне помочь, пожалуйста?

Answer 1

Используя функцию прямоугольника cv2,

(locations, preditions) = detect_and_predict_class(test_image, model) # make predictions from your model

for (box, pred) in zip(locationss, predictions):

    (startX, startY, endX, endY) = box # box coordinates returned from your model's predictions
    
    cv2.rectangle(input_image, (startX, startY), (endX, endY), color, 2) # color is the color of the bounding box you would like & 2 is the thickness of the bounding box

Ссылка на пример примера:

Компьютерщики для компьютерных фанатов пример использования

Документация по прямоугольнику CV2