颜色分类leetcode colorization pytorch PyTorch重新实现交互式深度着色

颜色分类leetcode PyTorch中的交互式深度着色| | | |具有学习深度先验的实时用户引导图像着色。 *, , , Angela S. Lin, Tianhe Yu,和.在ACM图形交易(SIGGRAPH 2017)中。这是我们用于交互式图像着色的PyTorch重新实现，由和编写。此存储库包含训练用法。原始的官方GitHub存储库（带有交互式GUI，最初是Caffe后端）是.官方仓库已经更新，支持后端的PyTorch模型，可以在这个仓库中训练。先决条件Linux或macOS Python 2或3 CPU或NVIDIA GPU + CUDA CuDNN入门安装安装PyTorch 0.4+和torchvision以及其他依赖项（例如，和）。您可以通过以下方式安装所有依赖项pip install -r requirements.txt克隆这个repo： git clone https://github.com/richzhang/colorization-pytorch cd colo