twishabansal / image-colourisation Goto Github PK

• Pytorch for building the model
• Streamlit for building the web application
• Heroku for deploying the web application

The path files for the models trained on landscapes, people, fruits, and animals are available as landscapes.pth, people.pth, fruits.pth and animals.pth.

1. Clone the repository with git clone https://gitlab.com/twishabansal/image-colourisation.git
2. Open image-colourization-starter.ipynb.
3. To load a particular path file in your notebook, run -

def load_checkpoint(filepath): 
    model = Encoder_Decoder()
    checkpoint = torch.load(filepath)
    model.load_state_dict(checkpoint['state_dict'])
    
    return model

model = load_checkpoint(filepath)

1. Clone the repository with git clone https://gitlab.com/twishabansal/image-colourisation.git
   
2. Documented Code for the model is available in the repository as image-colourization-starter.ipynb as an IPython notebook.
   
3. Refer to the code written to process the data, define the model, train it, and finally get a prediction.

The following datasets were used to train the respective models-

1. ImageNet
2. Flickr
3. Landscape Classification
4. Scene Classification

1. Clothes Classification
2. CelebA Dataset

1. Fruits360
2. Fruit Recognition

1. Animals10
2. Arthropod Taxonomy Orders Object Detection Dataset
3. Animal Faces
4. African Wildlife
5. Animals Dataset
6. The Oxford-IIIT Pet Dataset

1. Clone the repository with https://github.com/Priyansi/image-colouriser-streamlit.git
2. To install Streamlit - pip install streamlit
3. To run the app on http://localhost:8501 run streamlit run app/app.py

1. Colorful Image Colorization
2. Applications of AutoEncoders - Image Colourisation

All rights reserved. Licensed under the MIT License.