目录
文件夹批量resize
torch是最快的,能快1ms
分别是5ms 9ms 6ms,但是torch 占用cpu很高。
python opencv代码:
# -*- coding: utf-8 -*-
import glob
import cv2
import os
import time
def resize(dir_path,to_dir,to_size,to_shape):
os.makedirs(to_dir, exist_ok=True)
g = os.walk(dir_path)
img_files = ['%s/%s' % (i[0], j) for i in g for j in i[-1] if j.endswith(( 'png', 'jpeg', 'JPG'))]
# img_files = ['%s/%s' % (i[0], j) for i in g for j in i[-1] if j.endswith(('jpg', 'png', 'jpeg', 'JPG'))]
index = 0
for file in img_files:
img = cv2.imread(file) #