This commit is contained in:
Nataniel Ruiz
2019-12-25 20:43:31 -04:00
parent eec72da9c1
commit d70e90e898
+2
View File
@@ -33,6 +33,7 @@ from models import create_model
from util.visualizer import save_images
from util import html
import torch
import time
if __name__ == '__main__':
opt = TestOptions().parse() # get test options
@@ -84,6 +85,7 @@ if __name__ == '__main__':
if i % 5 == 0: # save images to an HTML file
print('processing (%04d)-th image... %s' % (i, img_path))
save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio, width=opt.display_winsize)
time.sleep(0.5)
webpage.save() # save the HTML