import numpy as np import pylib as py import tensorflow as tf import tf2lib as tl import data import module # ============================================================================== # = param = # ============================================================================== py.arg('--experiment_dir') py.arg('--batch_size', type=int, default=32) test_args = py.args() args = py.args_from_yaml(py.join(test_args.experiment_dir, 'settings.yml')) args.__dict__.update(test_args.__dict__) # ============================================================================== # = test = # ============================================================================== # data A_img_paths_test = py.glob(py.join(args.datasets_dir, args.dataset, 'testA'), '*.jpg') B_img_paths_test = py.glob(py.join(args.datasets_dir, args.dataset, 'testB'), '*.jpg') A_dataset_test = data.make_dataset(A_img_paths_test, args.batch_size, args.load_size, args.crop_size,
) py.arg('--test_label_path', default='./data/img_celeba/test_label.txt') py.arg('--test_att_names', choices=data.ATT_ID.keys(), nargs='+', default=['Bangs', 'Mustache']) py.arg('--test_ints', type=float, nargs='+', default=2) py.arg('--experiment_name', default='default') args_ = py.args() # output_dir output_dir = py.join('output', args_.experiment_name) # save settings args = py.args_from_yaml(py.join(output_dir, 'settings.yml')) args.__dict__.update(args_.__dict__) # others n_atts = len(args.att_names) if not isinstance(args.test_ints, list): args.test_ints = [args.test_ints] * len(args.test_att_names) elif len(args.test_ints) == 1: args.test_ints = args.test_ints * len(args.test_att_names) sess = tl.session() sess.__enter__() # make default # ============================================================================== # = data and model = # ==============================================================================
default='./data/zappos_50k/test_label_subcategories.txt') py.arg('--test_att_name', choices=data.ATT_ID.keys(), default='Heels') py.arg('--test_int_min', type=float, default=-2) py.arg('--test_int_max', type=float, default=2) py.arg('--test_int_step', type=float, default=0.5) py.arg('--experiment_name', default='default') args_ = py.args() # output_dir output_dir = os.path.join(args_.flask_path, py.join('output', args_.experiment_name)) # output_dir = py.join('output', args_.experiment_name) # save settings args = py.args_from_yaml(py.join(output_dir, 'settings_subcategories.yml')) args.__dict__.update(args_.__dict__) # others n_atts = len(args.att_names) sess = tl.session() sess.__enter__() # make default # ============================================================================== # = data and model = # ============================================================================== # data flask_img_dir = os.path.join(args.flask_path, args.img_dir) flask_test_label_path = os.path.join(args.flask_path, args.test_label_path)