← all repositories

richzhang/colorization-pytorch

A PyTorch implementation of a convolutional neural network for real-time user-guided image colorization.

602 stars Python Computer VisionML Frameworks
colorization-pytorch
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository provides a PyTorch reimplementation of an interactive deep colorization system that colorizes grayscale images using learned deep priors. The model uses convolutional neural networks trained on ImageNet to produce photorealistic colorizations in real time. It includes the training code, model definitions, and dataset preparation scripts.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.