Downloads · 30 days
0
image-colorization
davidramos/image-colorization
image-colorization is a machine learning model from davidramos. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
An implementation of the Pix2Pix paper for image colorization using PyTorch.
Access
Public
Updated Dec 1, 2024
Repo size
229 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pth229 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Dec 1, 2024
- Updated
- Dec 1, 2024
- SHA
- 15312813
Datasets
Model card
From the Hugging Face model README
An implementation of the Pix2Pix paper for image colorization using PyTorch.
Try a trained on Google Colab:
<a href="https://colab.research.google.com/github/DavidRamosArchilla/Colorizing-images/blob/main/image_colorization_demo.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
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))
- License
- mit
- Created
- Dec 1, 2024
- Updated
- Dec 1, 2024