python常用程式碼整理

太一吾鱼水發表於2024-03-25

1. 黑白圖、三通道圖轉化為單通道圖

第一種辦法(讀取時就轉化):img = cv2.imread('test.jpg', 0) # 引數0等價於:cv2.IMREAD_GRAYSCALE。 如果不加0,會預設讀取為三通道影像,即便原影像是灰度圖。
第二種辦法(讀取後再轉化):img = cv2.cvtColor(img , cv2.COLOR_RGB2GRAY)

PIL的辦法(讀取時就轉化): img = Image.open('test.jpg').convert('L') # 引數為"1"時示意轉化為二值化圖(位深度為1,每個畫素點的值為0或1)

2.讀寫txt

    data_path = '../../data'
    f_Edges = data_path + '/{}{}'.format('lines', '_Edge.txt')
    with open(f_Edges, 'w') as file:
        for a, l, lc in zip(axes, lines, line_colors):
            for i in range(len(l)):
                #寫入一條記錄
                file.writelines('{} {} {} {}\n'.format(l[i, 0, 0], l[i, 1, 0], l[i, 0, 1], l[i, 1, 1]))

相關文章