Найдите ROIS из маски R-CNN и нарисуйте прямоугольник через cv2.rectangle на изображении, чтобы узнать размер - PullRequest
0 голосов
/ 25 февраля 2020

Я пытаюсь нарисовать прямоугольник на изображении рубашки, из которого мне нужно выяснить контур и средние точки, чтобы узнать размер рубашки. Кто-нибудь может мне помочь, пожалуйста

#import libraries
from google.colab.patches import cv2_imshow
from scipy.spatial import distance as dist
from imutils import perspective
from imutils import contours
import numpy as np
import argparse
import imutils
import cv2


# Test on images present in testImages folder
testimagepath = os.path.join(ROOT_DIR, 'Dataset/testImages/roi')
testpath = os.path.join(ROOT_DIR, 'test')


# Detect from the original image
image = cv2.imread(os.path.join(testimagepath, i))
image = cv2.resize(image,( 1024, 1024))
results = model.detect([image], verbose=1)
#Getting data of first feature
r = results[0]

#showing image before roi
cv2_imshow(image)
# Getting roi and trying to draw rectangle
roi_arr=r['rois'][0]
cv2.rectangle(image, (roi_arr[0],roi_arr[1]), (roi_arr[2], roi_arr[3]), (255,0,0), 2)
cv2_imshow(image)
...