1234567891011121314151617181920212223242526272829303132 |
- import argparse
- import datetime
- import os
- import uuid
- import cv2
- def rename(img_dir, save_dir):
- filelist = os.listdir(img_dir)
- total_num = len(filelist)
- i = 1
- for item in filelist:
- time = datetime.datetime.now()
- t = str(time.year).zfill(4)+'-'+str(time.month).zfill(2) + '-'+str(time.day).zfill(2)
- img = cv2.imread(img_dir + '/' + item)
- if item.endswith('.jpg') or item.endswith('.png'):
-
- dst = save_dir + '/id_' + item.split('.')[0] + '.jpg'
- cv2.imwrite(dst, img)
- i = i + 1
- print('total %d to rename' % total_num)
- if __name__ == '__main__':
- parser = argparse.ArgumentParser()
- parser.add_argument('--img_dir', type=str)
- parser.add_argument('--save_dir', type=str)
- args = parser.parse_args()
- if not os.path.exists(args.save_dir):
- os.makedirs(args.save_dir)
- rename(args.img_dir, args.save_dir)
|