Install all the required dependencies using :
pip install requirements.txtThen clone the repository using :
git clone https://github.com/karthikeya-aduri/image-colorization
Then open terminal inside image-colorization folder/directory and run this command for running the application.
streamlit run app.py- The CNN model is trained on CIFAR-10. (It is still in experimental stages)
- The autoencoder model is trained on a landscape images dataset.
- [Landscape Images Dataset] (https://www.kaggle.com/datasets/theblackmamba31/landscape-image-colorization)
Autoencoder result :
