From 7212499f839db946c9296b84d6491fe5d6097dea Mon Sep 17 00:00:00 2001 From: junyanz Date: Tue, 13 Jun 2017 00:10:04 -0700 Subject: predict single image --- test.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index d2910ab..f019d10 100644 --- a/test.py +++ b/test.py @@ -1,18 +1,17 @@ import time import os from options.test_options import TestOptions -opt = TestOptions().parse() # set CUDA_VISIBLE_DEVICES before import torch - from data.data_loader import CreateDataLoader from models.models import create_model from util.visualizer import Visualizer from pdb import set_trace as st from util import html -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 +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() -- cgit v1.2.3-70-g09d2