Python 并行库 joblib

import matplotlib.pyplot as plt # 读取图片
from joblib import Parallel, delayed # 并行计算
from rich.progress import track # 可视化进度条


def read_img(path):
    return plt.imread(path)

imgs = Parallel(n_jobs=-1)(
    delayed(read_img)(path)
    for path in track(csv["im_name"], description="Loading images ... ", transient=True)
    # transient=True 指定进度条在完成后隐藏
) # parallel() 结束之后，imgs 是 List[np.ndarray]

imgs = np.asarray(imgs) # 转化为 np.ndarray