This an implementation of Undercomplete Autoencoder which is used for converting an RGB image to an GRAY scale image. The autoencoder is implemented using the TensorFlow framework in Python.
- TensorFlow==1.11.0
- Python==3.5.6
- Numpy==1.15.2
- OpenCV==3.4.3
Pre-trained model is present in saved_model.zip. Load the saved model to skip the training.
NOTE: Do check for path errors in the code, modify it according to your need. Eg. The saver.train("....") has a path that must be changed before running the session.
- Thanks to people at www.flicker.com for sharing the dataset.