An implementation of the Pix2Pix paper for image colorization using PyTorch.

Try a trained on Google Colab:

Open In Colab


To download it:

from huggingface_hub import hf_hub_download

model_path = hf_hub_download(
    repo_id="davidramos/image-colorization", 
    filename="model.pth" 
)

and to load the weights into a model, first clone this repo (or grab the needed files):

git clone https://github.com/yourusername/colorizing_images.git
cd colorizing_images

and create a model intance:

from models import ColorizingNet
model = ColorizingNet(device)
model.load_state_dict(torch.load(model_path))
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference API
Unable to determine this model's library. Check the docs .

Dataset used to train davidramos/image-colorization