interactive-deep-colorization

Why we included this project

Photographers and archivists working through large collections of old or monochrome photos have a concrete use for this. The colorizer is interactive rather than fully automatic: you drop a few colored strokes or click hints, and a learned deep prior from the authors' SIGGRAPH 2017 paper fills the rest of the frame with consistent color, which is what you want when accuracy or historical plausibility matters. The repo bundles a reference model, a barebones Jupyter notebook for quick experiments, and the full Qt-based GUI from the paper, where hint points can be repositioned and recolored as you work. It runs on Caffe or PyTorch and handles CPU or GPU, so you can test the workflow cheaply before committing to a production setup. The approach also shipped as a colorize feature in Adobe Photoshop Elements 2020, a decent sign that the interaction model holds up outside the lab.