OpenCV Python изменить размер изображения - PullRequest
Новая
       22

OpenCV Python изменить размер изображения

0 голосов
/ 11 июня 2018

Я пытаюсь изменить размер контура до 28x28 пикселей и пропустить его через мою модель, чтобы обнаружить цифру и отобразить ее.Однако у меня проблема с функцией cv2.resize().Вот мой код: 

ret, frame = cap.read()

if ret == False:
    continue
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
blurred = cv2.GaussianBlur(gray, (5, 5), 0)
edged = cv2.Canny(blurred, 50, 200, 255)

# Threshold the image
threshold = cv2.adaptiveThreshold(edged, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, 11, 4)

# Find contours in the image
_, contours, _= cv2.findContours(threshold, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)


# ******************************************************************* #
# Get rectangles contains each contour
rects = [cv2.boundingRect(ctr) for ctr in contours]

for rect in rects:
    # Draw the rectangles
    cv2.rectangle(frame, (rect[0], rect[1]), (rect[0] + rect[2], rect[1] + rect[3]), (0, 255, 0), 3) 

    # Make the rectangular region around the digit
    leng = int(rect[3] * 1.6)
    pt1 = int(rect[1] + rect[3] // 2 - leng // 2)
    pt2 = int(rect[0] + rect[2] // 2 - leng // 2)

    height, width = frame.shape[:2]

    if pt1+leng > width or pt2+leng > height:
        continue;

    roi = frame[pt1:pt1+leng, pt2:pt2+leng]

    # Resize the image
    roi = cv2.resize(roi, (28, 28), cv2.INTER_AREA)
    roi = cv2.dilate(roi, (3, 3))

    #nbr = clf.predict(np.array([roi_hog_fd], 'float64'))
    #cv2.putText(frame, str(int(nbr)), (rect[0], rect[1]),cv2.FONT_HERSHEY_DUPLEX, 1, (0, 0, 255), 3)
    cv2.putText(frame, "2", (rect[0], rect[1]),cv2.FONT_HERSHEY_DUPLEX, 1, (0, 0, 255), 3)

Я получаю ошибку:

Ошибка: C: \ projects \ opencv-python \ opencv \ modules \ imgproc \ src \ resize.cpp: 4044: ошибка: (-215) ssize.width> 0 && ssize.height> 0 в функции cv :: resize

Изменение размера исходного кадра видео (без применения какой-либо другой функции cv2на это раньше) все хорошо.

Есть предложения?

Ответы [ 2 ]

0 голосов
/ 14 июня 2018

Проблема заключалась в том, что я пытался получить координатную картинку, которая была бы отрицательной, поэтому он не мог ее получить.Я просто расширил условие if pt1+leng > width or pt2+leng > height or pt2 < 0 or pt1 < 0: и оно работает.

0 голосов
/ 11 июня 2018

Попробуйте создать копию roi, прежде чем пытаться изменить размер, 

roiImg = roi.copy()
roiImg = cv2.resize(roiImg, (28, 28), cv2.INTER_AREA)
...