diff --git a/cyclegan/test.py b/cyclegan/test.py index a8a5ab5..6cf1431 100644 --- a/cyclegan/test.py +++ b/cyclegan/test.py @@ -86,7 +86,10 @@ 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) + + # Print metrics + print('{} images. L1 error: {}. L2 error: {}. L0 error: {}. L_-inf error: {}. Perceptual error: {}.'.format(n_samples, + l1_error / n_samples, l2_error / n_samples, l0_error / n_samples, min_dist / n_samples, perceptual_error / n_samples)) webpage.save() # save the HTML