【Python脚本】显示试验可视化对比进行拼接

加载实验中可视化的图片后，对不同方法进行对比，对多个图像进行横向拼接，保存图片进行对比。
代码如下

import os
import cv2
from tqdm import tqdm
import numpy as np

array_of_img = [] 

def read_directory(directory_name, data_dir):
    for filename in tqdm(os.listdir(data_dir)):
        img1 = cv2.imread(directory_name + "/xvo_kitti_test/vis/" + filename)
        img2 = cv2.imread(directory_name + "/sdp_kitti_test/vis/" + filename)
        img3 = cv2.imread(data_dir + "/" + filename)

        img1 = cv2.resize(img1,dsize=(416,128),fx=1,fy=1,interpolation=cv2.INTER_LINEAR)
        img2 = cv2.resize(img2,dsize=(416,128),fx=1,fy=1,interpolation=cv2.INTER_LINEAR)
        img3 = cv2.resize(img3,dsize=(416,128),fx=1,fy=1,interpolation=cv2.INTER_LINEAR)

        heng = np.hstack((img3, img1, img2))
        cv2.imwrite(directory_name + "/compara_result/" + filename, heng)

def main():
    img_dir = "/root/program/demo/output"
    data_dir = "/root/program/data/testing/color"
    read_directory(img_dir, data_dir)

if __name__ == '__main__':
    main()