（信号灯三）三格子信号灯三等分截取后统一尺寸

构建代码

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import cv2

# 切割
def get_three_box(img_shape):
    # 竖向三座信号灯
    if img_shape[0] > 2*img_shape[1]:
        a = (img_shape[1], img_shape[0]) # 横向尺寸, 纵向尺寸
        circle_point = (int(a[0]/2),int(a[1]/2))
        x_inch = int(a[0]*0.45)
        y_inch = int(a[1]*0.14)
        dy = int(a[1]*0.015)
        box_middle = (circle_point[0]-x_inch, circle_point[1]-y_inch+2*dy, circle_point[0]+x_inch, circle_point[1]+y_inch+2*dy)
        box_up = (box_middle[0], 4*dy, box_middle[2], box_middle[1])
        box_down = (box_middle[0], box_middle[3], box_middle[2], a[1]-4*dy)
        
        return box_up, box_middle, box_down
    # 横向三座信号灯
    else:
        a = (img_shape[1], img_shape[0]) # 横向尺寸, 纵向尺寸
        circle_point = (int(a[0]/2),int(a[1]/2))
        x_inch = int(a[0]*0.14)
        y_inch = int(a[1]*0.45)
        dx = int(a[0]*0.015)
        box_middle = (circle_point[0]-x_inch+6*dx, circle_point[1]-y_inch, circle_point[0]+x_inch+2*dx, circle_point[1]+y_inch)
        box_up = (6*dx, box_middle[1], box_middle[0], box_middle[3])
        box_down = (box_middle[2], box_middle[1], a[0]-6*dx, box_middle[3])
        
        return box_up, box_middle, box_down

# 截取
def cut_three_box_from_tl(img):
    box_up, box_middle, box_down = get_three_box(img.shape)
    img_up = img[box_up[1]:box_up[3], box_up[0]:box_up[2]]
    img_middle = img[box_middle[1]:box_middle[3], box_middle[0]:box_middle[2]]
    img_down = img[box_down[1]:box_down[3], box_down[0]:box_down[2]]
    return img_up, img_middle, img_down

# resize到相同尺寸
def resize_three_image(img, height, width):
    # get box
    box_up, box_middle, box_down = get_three_box(img.shape)
    # cut
    img_up = img[box_up[1]:box_up[3], box_up[0]:box_up[2]]
    img_middle = img[box_middle[1]:box_middle[3], box_middle[0]:box_middle[2]]
    img_down = img[box_down[1]:box_down[3], box_down[0]:box_down[2]]
    # resize
    img_up = cv2.resize(img_up,(height,width), interpolation=cv2.INTER_CUBIC)
    img_middle = cv2.resize(img_middle,(height,width), interpolation=cv2.INTER_CUBIC)
    img_down = cv2.resize(img_down,(height,width), interpolation=cv2.INTER_CUBIC)
    
    return img_up, img_middle, img_down

执行代码

img = mpimg.imread('./test_images/100.jpg')
img_up, img_middle, img_down = cut_three_box_from_tl(img)
img_up1, img_middle1, img_down1 = resize_three_image(img, 20, 30)
fig = plt.gcf()
fig.set_size_inches(12, 7)

plt.subplot(241)
plt.imshow(img)
# plt.axis('off')
plt.title('original_img1')

plt.subplot(242)
plt.imshow(img_up)
# plt.axis('off')
plt.title('img_up')

plt.subplot(243)
plt.imshow(img_middle)
# plt.axis('off')
plt.title('img_middle')

plt.subplot(244)
plt.imshow(img_down)
# plt.axis('off')
plt.title('img_down')

plt.subplot(246)
plt.imshow(img_up1)
# plt.axis('off')
plt.title('img_up1')

plt.subplot(247)
plt.imshow(img_middle1)
# plt.axis('off')
plt.title('img_middle1')

plt.subplot(248)
plt.imshow(img_down1)
# plt.axis('off')

执行结果