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)
