search

kwotsin / TensorFlow-ENet

TensorFlow implementation of ENet
MIT License
257 stars 123 forks source link

Added flags to predict_segmentation.py #16

Open davidtvs opened 6 years ago

davidtvs commented 6 years ago

Added a few flags to predict_segmentation.py (similar to train_enet.py and test_enet.py) so it can be more customizable.

Also included a predict.sh file with a usage example of the flags.