Нарисуйте прямоугольник с соотношением сторон 3: 2 на изображении

Question

Я пытаюсь нарисовать прямоугольник с соотношением сторон 3: 2.Я использую OpenCV для обнаружения объектов на изображении.Таким образом, из выходных значений я рисую прямоугольник с min X, min Y, max X, max Y. Теперь мне нужно, чтобы у этого прямоугольника было соотношение сторон 3: 2 от начальных точек, то есть min x и min Y.

Оно не должно выходить за пределы исходного изображения: max X и max Y, и прямоугольник не должен быть меньше существующего прямоугольника вокруг обнаруженных объектов.

Answer 1

Сначала вычислите новый размер блока, чтобы его w: h было 3: 2.И затем обрежьте прямоугольник, если одна из его сторон длиннее, чем у изображения.

После определения размера ящика мы рассчитываем центр ящика.По умолчанию центр прямоугольника остается тем же, но он будет смещен, если прямоугольник пересекает границу изображения.

Наконец, мы можем использовать размер прямоугольника и центр прямоугольника для вычисления координат углов прямоугольника.

import cv2

def draw_rectangle(img, min_x, min_y, max_x, max_y):
    # resize box to 3:2(only enlarge it)
    # determine the box_w and box_h
    box_w, box_h = max_x-min_x, max_y-min_y
    if box_w/box_h < 3/2:
        box_w = int(box_h*(3/2))
    else:
        box_h = int(box_w*(2/3))

    # trim the box so it won't be bigger than image
    h, w = img.shape[:2]
    box_w = w if box_w > w else box_w
    box_h = h if box_h > h else box_h

    # determine the center of box

    # the default box center
    box_center_x = (min_x+max_x)//2
    box_center_y = (min_y+max_y)//2

    # shift the box if it cross the boundary
    if box_center_x + box_w//2 > w:
        box_center_x = w - box_w//2
    elif box_center_x - box_w//2 < 0:
        box_center_x = box_w//2
    if box_center_y + box_h//2 > h:
        box_center_y = h - box_h//2
    elif box_center_y - box_h//2 < 0:
        box_center_y = box_h//2

    # calculate the corner of the box
    min_x, max_x = box_center_x - box_w//2, box_center_x + box_w//2
    min_y, max_y = box_center_y - box_h//2, box_center_y + box_h//2
    cv2.rectangle(img, (min_x, min_y), (max_x, max_y), (255,0,0), thickness=10)
    return img

img = cv2.imread('image.jpg')
min_x, min_y, max_x, max_y = 0, 0, 400, 230
img = draw_rectangle(img, min_x, min_y, max_x, max_y)
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

Answer 2

Вот один из способов решить эту проблему:

# determine the y / x length of the rectangle
len_X = maxX - minX
len_Y = maxY - minY

# determine the largest side, this will be the 3 in the aspect ratio
if len_X > len_Y:
    # check if the shorter side is larger than a 3:2 ration
    if len_Y > len_X * (3/2):
        # if so, increase larger side to 3:2 ratio
        len_X = len_Y * 1.5
    else:
        # else, increase shorter side to 3:2 ratio
        len_Y = len_X * (3/2) 
else:
    # same as above
    if len_X > len_Y * (3/2):
        len_Y = len_X * 1.5
    else:
        len_X = len_Y * (3/2) 

# if the rectangle exceeds the image, constrain the rectangle
# other option (commented): move the starting position
if minX + len_X > img.shape[1]:
    len_X = img.shape[1]-minX
    #minX = img.shape[1]-len_X
if minY + len_Y > img.shape[0]:
    len_Y = img.shape[0]-minY
    #minY = img.shape[0]-len_Y

# draw the rectangle
cv2.rectangle(img, (minX, minY), (minX + len_X, minY + len_Y), (0,0,255),1)