python 多进程保存图片，速度加倍

 Python中的多线程和多进程有所区别，这里采用多进程，作为样例么有用进程池，而是手动创建两个进程，用于保存列表中的图片

单个进程的耗时： 21.24s

两个进程的耗时：10.69s

用了两个进程，耗时减半，实际应用中可以，用进程池的Queue的方式时间进程间信息共享，利用lock保证数据的安全，加lock会增加耗时，需要妥善处理

import queue
import threading
import os
import cv2
import time
from multiprocessing import Process

def read_and_save_img():
    src_dir = r"/home/cody/Documents/mygot10k/train/GOT-10k_Train_010000"
    dst_dir = r"/home/cody/Documents/trampoline/temp"

    imgs = os.listdir(src_dir)
    imgs = [x for x in imgs if x.endswith(".jpg")]
    imgs = [os.path.join(src_dir,x) for x in imgs]

    time_r = time.time()
    list_imgs = []
    for i in range(len(imgs)):
        img_ori = cv2.imread(imgs[i])
        list_imgs.append(img_ori)
    print("read time", time.time() - time_r)
    
    time_s = time.time()
    for i in range(len(imgs)):
        cv2.imwrite(os.path.join(dst_dir,str(i).rjust(3,"0")+".jpg"),list_imgs[i])
    print(len(imgs))
    print("write time",time.time() - time_s)

def save_img(list_imgs,dst_dir):
    for i in range(len(list_imgs)):
        cv2.imwrite(os.path.join(dst_dir,str(i).rjust(3,"0")+".jpg"),list_imgs[i])


def read_and_save_img_process():
    src_dir = r"/home/cody/Documents/mygot10k/train/GOT-10k_Train_010000"
    dst_dir = r"/home/cody/Documents/trampoline/temp"

    time_r = time.time()
    imgs = os.listdir(src_dir)
    imgs = [x for x in imgs if x.endswith(".jpg")]
    imgs = [os.path.join(src_dir, x) for x in imgs]

    list_imgs = []
    for i in range(len(imgs)):
        img_ori = cv2.imread(imgs[i])
        list_imgs.append(img_ori)
    print("read time",time.time() - time_r)
    

    time_s = time.time()
    ran = int(len(list_imgs) * 0.5)

    ps1 = Process(target=save_img, args=(list_imgs[:ran], dst_dir))
    ps2 = Process(target=save_img, args=(list_imgs[ran:], dst_dir))
    ps1.start()
    ps2.start()
    ps1.join()
    ps2.join()

    print("write time",time.time() - time_s)


if __name__ == '__main__':
    # read_and_save_img() # 892,21.24644923210144
    read_and_save_img_process() #10.695387125015259