....
Testing:
* Run python test.py --image examples/input.png --mask examples/mask.png --output examples/output.png --checkpoint model_logs/your_model_dir
.
Visualization on TensorBoard for training and validation is supported. Run tensorboard --logdir model_logs --port 6006
to view training progress.