bhaecker

I used black and white images to train, which means that I had to change the channel number to 1 in input_layer, line 19 in save_model.py: ``` input_layer = tf.keras.layers.Input([FLAGS.input_size,...