1. '''
    2. author:tonyzhang
    3. function:transfrom img to size H W and save
    4. '''
    5. import cv2
    6. import os
    7. import glob
    9. dir_path = "F:/SPAQImage/*.jpg"
    10. save_path = "F:/SPAQImage224/"
    12. for i in glob.glob(dir_path):
    13.     img = cv2.imread(i)
    14.     print(i)
    15.     img = cv2.resize(img,(224,224))
    16.     cv2.imwrite(os.path.join(save_path,os.path.basename(i)),img)

    上面的单线程版本，太慢了
    下面搞了个多线程，快了很多，cpu利用率拉满

    1. '''
    2. author:tonyzhang
    3. function:transfrom img to size H W and save
    4. '''
    5. import cv2
    6. import os
    7. import glob
    8. import concurrent.futures
    9. from threading import Thread
    11. num_threads = 10
    12. dir_path = "T:/PARA/img_all/"
    13. save_path = "T:/PARA/img_all2/"
    15. def process_image(filelist):
    16.     for filename in filelist:
    17.         print(filename)
    18.         file = os.path.join(dir_path,filename)
    19.         img = cv2.imread(file)
    20.         img = cv2.resize(img,(224,224))
    21.         cv2.imwrite(os.path.join(save_path,filename),img) 
    27. def multi_thread_process(dir_path):
    28.     if num_threads == 1:
    29.         process_image(os.listdir(dir_path))
    30.         return
    31.     filelist_total = os.listdir(dir_path)
    32.     filenum = len(filelist_total)
    33.     filenum_each_thread = int(filenum/num_threads)
    34.     thread_list = []
    35.     for i in range(num_threads-1):
    36.         thread_list.append(Thread(target=process_image, args=(filelist_total[i*filenum_each_thread : (i+1)*filenum_each_thread],)))
    37.     thread_list.append(Thread(target=process_image, args=(filelist_total[(num_threads-1)*filenum_each_thread :],)))
    38.     for th in thread_list:
    39.         th.start()
    40.     for th in thread_list:
    41.         th.join()
    43. multi_thread_process(dir_path)