图片移到根目录
import os
import shutil
import timeimport cv2class TimeStr:def __init__(self):self.last_time, self.counter = '', 0def get_time(self, fmt="%m%d_%H%M_%S"):now_time = time.strftime(fmt)self.counter += 1if now_time != self.last_time:self.last_time, self.counter = now_time, 0return f"{now_time}_{self.counter}"def get_time_sec(self):return self.get_time("%m%d_%H%M_%S")def get_time_mm(self):return self.get_time("%m%d_%H%M")if __name__ == '__main__':name_get=TimeStr()dir_base=r"D:\data\wo_bei\train\1_err"save_dir=r"D:\data\wo_bei\train\1_err_jpg/"os.makedirs(save_dir,exist_ok=True)dir_base=dir_base.replace("\\", "/")save_dir=save_dir.replace("\\", "/")img_files = ['%s/%s' % (i[0].replace("//", "/"), j) for i in os.walk(dir_base) for j in i[-1] ifj.lower().endswith(('mp4', 'png', 'jpg'))]for img_path in img_files:save_path = save_dir+os.path.basename(img_path)if os.path.exists(save_path):save_path=os.path.join(save_dir,name_get.get_time_mm())+".jpg"if img_path.lower().endswith('.png'):img = cv2.imread(img_path)cv2.imwrite(save_path, img)else:shutil.move(img_path, save_path)else:if img_path.lower().endswith('.png'):img = cv2.imread(img_path)cv2.imwrite(save_path, img)else:shutil.move(img_path, save_dir)
