|
import os |
|
from options.test_options import TestOptions |
|
from data import CreateDataLoader |
|
from models import create_model |
|
from util.visualizer import save_images |
|
from util import html |
|
|
|
|
|
if __name__ == '__main__': |
|
opt = TestOptions().parse() |
|
opt.num_threads = 1 |
|
opt.batch_size = 1 |
|
opt.serial_batches = True |
|
opt.no_flip = True |
|
opt.display_id = -1 |
|
data_loader = CreateDataLoader(opt) |
|
dataset = data_loader.load_data() |
|
model = create_model(opt) |
|
model.setup(opt) |
|
|
|
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),reflesh=0, folder=opt.imagefolder) |
|
if opt.test_continuity_loss: |
|
file_name = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch), 'continuity.txt') |
|
file_name1 = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch), 'continuity-r.txt') |
|
if os.path.exists(file_name): |
|
os.remove(file_name) |
|
if os.path.exists(file_name1): |
|
os.remove(file_name1) |
|
|
|
|
|
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() |
|
|
|
|
|
save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio, width=opt.display_winsize) |
|
|
|
webpage.save() |
|
if opt.model == 'regressor': |
|
print(model.cnt) |
|
print(model.value/model.cnt) |
|
print(model.minval) |
|
print(model.avg/model.cnt) |
|
print(model.max) |
|
html = os.path.join(web_dir,'cindex'+opt.imagefolder[6:]+'.html') |
|
f=open(html,'w') |
|
print('<table border="1" style=\"text-align:center;\">',file=f,end='') |
|
print('<tr>',file=f,end='') |
|
print('<td>image name</td>',file=f,end='') |
|
print('<td>realA</td>',file=f,end='') |
|
print('<td>realB</td>',file=f,end='') |
|
print('<td>fakeB</td>',file=f,end='') |
|
print('</tr>',file=f,end='') |
|
for info in model.info: |
|
basen = os.path.basename(info[0])[:-4] |
|
print('<tr>',file=f,end='') |
|
print('<td>%s</td>'%basen,file=f,end='') |
|
print('<td><img src=\"%s/%s_real_A.png\" style=\"width:44px\"></td>'%(opt.imagefolder,basen),file=f,end='') |
|
print('<td>%.4f</td>'%info[1],file=f,end='') |
|
print('<td>%.4f</td>'%info[2],file=f,end='') |
|
print('</tr>',file=f,end='') |
|
print('</table>',file=f,end='') |
|
f.close() |
|
|