文件处理python小程序——对比图片并拼接（python+opencv）

# JackHu, 2021
import os 
import cv2
import numpy as np
image_dir_list = ['/mnt/ScaledYOLOv4-middle-all/output','/mnt/ScaledYOLOv4-middle-all-focalv1/output']
image_name_list = ['scv4','scv4focal']
prefix = 'gt'
for i in range(len(image_name_list)):
    prefix +='_' + image_name_list[i]
n = len(image_dir_list)
m = 888
image_list = os.listdir(image_dir_list[1])  #第二个文件夹中的所有图片为最大公约数集
# nummm = 0
# for _ in image_list:
#     if not _.endswith('.jpg'):
#         image_list.remove(_)
#         nummm +=1
# print('removed %d rubbish!'%nummm)
gt_dir = '/mnt/data-ractified/all/ground-truth/results'
save_path = '/mnt/compare-results'
for i in range(m):
    image = image_list[i]
    flag = 0
    print('start'+image)
    gt_image = gt_dir + '/' +image_list[i]
    gt_file = cv2.imread(gt_image)
    row = gt_file.shape[0]
    print(row)
    column = gt_file.shape[1]
    print(column)
    init = cv2.imread(gt_image)
    print('gt finished'+image)
    for dir in image_dir_list:
        image_path = dir+ '/' +image
        im_to_combine = cv2.imread(image_path)
        init = np.concatenate([init, im_to_combine], 1)
        print('finished'+dir+image)
    cv2.imwrite(save_path + '/'+prefix+image,init)
    print('save' + image)