| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- import os
- import sys
- import cv2
- import numpy as np
- from vi_core.img_resize import resize
- from_dir = sys.argv[1]
- to_dir = from_dir
- resize_param = 1
- if len(sys.argv) == 4:
- to_dir = sys.argv[2]
- resize_param = float(sys.argv[3])
- else:
- resize_param = float(sys.argv[2])
- from_class_dirs = os.listdir(from_dir)
- class_num = len(from_class_dirs)
- for i in range(class_num):
- from_class_dir = from_class_dirs[i]
- print('{}/{} {}'.format(i+1, class_num, from_class_dir))
- to_class_dir = os.path.join(to_dir, from_class_dir)
- if not os.path.exists(to_class_dir):
- os.makedirs(to_class_dir)
- from_names = os.listdir(os.path.join(from_dir, from_class_dir))
- from_names = [n for n in from_names if n.lower().endswith('.jpg')]
- name_num = len(from_names)
- for j in range(name_num):
- from_name = from_names[j]
- basename, ext = os.path.splitext(from_name)
- from_jpg = os.path.join(from_dir, from_class_dir, from_name)
- to_jpg = os.path.join(to_class_dir, from_name)
- im_inp = cv2.imread(from_jpg)
- if resize_param > 1:
- im_inp, _, _ = resize(im_inp, int(resize_param), int(resize_param), 1)
- elif resize_param < 1:
- im_inp, _, _ = resize(im_inp, -1, -1, float(resize_param))
- cv2.imwrite(to_jpg, im_inp)
- print(' {}/{} {}'.format(j+1, name_num, from_name))
-
|
