def per_step(img, e, s): plt.subplot(1, 3, 1) ImageUtils.imshow(transferer.content_image, 'Content Image') plt.subplot(1, 3, 2) ImageUtils.imshow(img, f'New Image, epoch {e}, step {s}') plt.subplot(1, 3, 3) ImageUtils.imshow(transferer.style_image, 'Style Image') ImageUtils.flashplot() fig = plt.figure(1) fig.clf() ax = fig.subplots(nrows=2, ncols=1)
) plotset2 = ( # This shows 1x3 simple layout ((1, 3, 1), content_image, 'Content Image'), ((1, 3, 2), stylized_image, 'New Image'), ((1, 3, 3), style_orig, 'Style Image'), ) for c, i, t in plotset1: plt.subplot(*c) ImageUtils.imshow(i, t) # View style transfers one by one input('Next? : ') ImageUtils.flashplot() print("\t> Finished") except Exception as e: print(" - Failed on image", i) print(f"\t> Image failed on {filename}: {e}.") fails += 1 # This skips fails. To be fixed later # For testing if fails > 5: quit() # if "n" == input(f" - Image failed on {filename}: {e}.\n - Type 'n' if to stop, anything else to go: "): # break input('Finish? : ')