forked from lengstrom/fast-style-transfer
/
resize_training_data.py
39 lines (31 loc) · 1.19 KB
/
resize_training_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import sys, os
sys.path.insert(0, 'src')
from argparse import ArgumentParser
from utils import save_img, get_img, exists, list_files
TRAIN_PATH = 'data/bin/train2014/train2014'
RESULT_PATH = 'data/train2014'
def build_parser():
parser = ArgumentParser()
parser.add_argument('--train-path', type=str,
dest='train_path', help='path to training images folder',
metavar='TRAIN_PATH', default=TRAIN_PATH)
parser.add_argument('--result-path', type=str,
dest='result_path', help='path to saved training images folder',
metavar='RESULT_PATH', default=RESULT_PATH)
return parser
def check_opts(opts):
exists(opts.train_path, "train path not found!")
exists(opts.result_path, "result path not found!")
def main():
parser = build_parser()
options = parser.parse_args()
check_opts(options)
files = list_files(options.train_path)
print('hihi')
for x in files:
in_path = os.path.join(options.train_path, x)
out_path = os.path.join(options.result_path, x)
img = get_img(in_path, (256,256,3))
save_img(out_path, img)
if __name__ == '__main__':
main()