An implementation of the Pix2Pix paper for image colorization using PyTorch.
Try a trained on Google 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))