From b77d9b593c9fa3b0db4e5e2a77c23777abac57ce Mon Sep 17 00:00:00 2001 From: junyanz Date: Tue, 27 Mar 2018 19:41:45 -0400 Subject: code refactoring --- test.py | 52 +++++++++++++++++++++++++++------------------------- 1 file changed, 27 insertions(+), 25 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index 863e550..8444bd9 100644 --- a/test.py +++ b/test.py @@ -1,32 +1,34 @@ import os from options.test_options import TestOptions -from data.data_loader import CreateDataLoader -from models.models import create_model +from data import CreateDataLoader +from models import create_model from util.visualizer import Visualizer from util import html -opt = TestOptions().parse() -opt.nThreads = 1 # test code only supports nThreads = 1 -opt.batchSize = 1 # test code only supports batchSize = 1 -opt.serial_batches = True # no shuffle -opt.no_flip = True # no flip -data_loader = CreateDataLoader(opt) -dataset = data_loader.load_data() -model = create_model(opt) -visualizer = Visualizer(opt) -# create website -web_dir = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch)) -webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch)) -# test -for i, data in enumerate(dataset): - if i >= opt.how_many: - break - model.set_input(data) - model.test() - visuals = model.get_current_visuals() - img_path = model.get_image_paths() - print('%04d: process image... %s' % (i, img_path)) - visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio) +if __name__ == '__main__': + opt = TestOptions().parse() + opt.nThreads = 1 # test code only supports nThreads = 1 + opt.batchSize = 1 # test code only supports batchSize = 1 + opt.serial_batches = True # no shuffle + opt.no_flip = True # no flip -webpage.save() + data_loader = CreateDataLoader(opt) + dataset = data_loader.load_data() + model = create_model(opt) + visualizer = Visualizer(opt) + # create website + web_dir = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch)) + webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch)) + # test + for i, data in enumerate(dataset): + if i >= opt.how_many: + break + model.set_input(data) + model.test() + visuals = model.get_current_visuals() + img_path = model.get_image_paths() + print('%04d: process image... %s' % (i, img_path)) + visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio) + + webpage.save() -- cgit v1.2.3-70-g09d2