step = 1 # Create logdir name logdir = "logs/{}-{}".format( os.path.basename(__file__), datetime.datetime.now().strftime("%Y-%m-%d_%H%M%S")) if not os.path.exists("logs"): os.mkdir("logs") # TF 1.6 will do this by itself network = Network() network.construct(logdir, ["original_greyscale", "original_rgb", "result"]) #visualize.copy_image(image_filename, "out_image_original.png") #visualize.clear_image("out_image.png") image_matrix = dataset.get_image_matrix_grayscale() image_matrix_rgb = dataset.get_image_matrix_rgb() #visualize.visualize_image_matrix_grayscale(image_matrix, "out_image_original_gray.png") #for i in range(10): network.visualize_image(image_matrix_rgb, "original_rgb") network.visualize_image(image_matrix, "original_greyscale") hopfield = Hopfield(image_matrix, alpha=alpha, strmost=strmost) while True: # recompute strmost _new_strmost = 1.0 * (step * batchsize - strmost_increase_after) / \ (strmost_increase_until - strmost_increase_after) * \