示例#1
0
# 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) * \
                           (strmost_final - strmost) + strmost