区域选择代码

偏执的太偏执、 2022-12-02 00:59 129阅读 0赞

import matplotlib.pyplot as plt
    import matplotlib.image as mpimg
    import numpy as np
    
    # 读取图片打印信息
    image = mpimg.imread('test.jpg')
    print('This image is: ', type(image), 
             'with dimensions:', image.shape)
    
    # Pull out the x and y sizes and make a copy of the image
    ysize = image.shape[0]
    xsize = image.shape[1]
    region_select = np.copy(image)
    
    # 定义感兴趣的三角区域
    # 请记住原点(x=0,y=0）位于图像左上方
    # Note:如果你运行代码，将会发现这是不可运行的值
    # 但是你很快会在测试中用到 
    left_bottom = [0, 539]
    right_bottom = [900, 300]
    apex = [400, 0]
    
    # 拟合线 (y=Ax+B) 来识别三面感兴趣的区域
    # np.polyfit() 返回合适的 [A, B] 的系数
    fit_left = np.polyfit((left_bottom[0], apex[0]), (left_bottom[1], apex[1]), 1)
    fit_right = np.polyfit((right_bottom[0], apex[0]), (right_bottom[1], apex[1]), 1)
    fit_bottom = np.polyfit((left_bottom[0], right_bottom[0]), (left_bottom[1], right_bottom[1]), 1)
    
    # 找到区域内部的线
    XX, YY = np.meshgrid(np.arange(0, xsize), np.arange(0, ysize))
    region_thresholds = (YY > (XX*fit_left[0] + fit_left[1])) & \
                        (YY > (XX*fit_right[0] + fit_right[1])) & \
                        (YY < (XX*fit_bottom[0] + fit_bottom[1]))
    
    # 彩色像素位于感兴趣区域内的红色
    region_select[region_thresholds] = [255, 0, 0]
    
    # 显示图像
    plt.imshow(region_select)
    
    # 如果没有显示，取消注释
    # plt.show()

呈现出的效果图  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0NDQ3J1bm5lcg_size_16_color_FFFFFF_t_70_pic_center]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0NDQ3J1bm5lcg_size_16_color_FFFFFF_t_70_pic_center]: /images/20221123/01e6e402f43a4083bb583537ed49d1d4.png