def finish(self): final_kernel = post_process_k(self.curr_k, n=self.conf.n_filtering) save_final_kernel(final_kernel, self.conf) print('KernelGAN estimation complete!') run_zssr(final_kernel, self.conf) print('FINISHED RUN (see --%s-- folder)\n' % self.conf.output_dir_path + '*' * 60 + '\n\n')
def eval(self): self.quick_eval() if self.conf.debug: self.plot() util.save_final_kernel(util.move2cpu(self.curr_k), self.conf) plt.imsave( os.path.join(self.conf.output_dir, '%s.png' % self.conf.abs_img_name), self.upsampled_img) if self.gt_img is not None: print('Upsampler PSNR = ', self.UP_psnrs[-1]) if self.gt_kernel is not None: print("Downsampler PSNR = ", self.DN_psnrs[-1]) print('*' * 60 + '\nOutput is saved in \'%s\' folder\n' % self.conf.output_dir) plt.close('all')