def show_image(image, title='', save_path=None): plt.figure() image = image.cpu().clone().squeeze(0) image = Processing.postprocessor(image) plt.imshow(image) plt.title(title) if save_path is not None: plt.savefig(save_path) plt.pause(0.001)
def show_image(image, title='', save_path=None): plt.figure() image = image.cpu().clone().squeeze(0) image = Processing.postprocessor(image) plt.imshow(image) plt.title(title) if save_path is not None: plt.savefig(save_path) plt.pause(0.001)