-
PyTorch实现的交互式深度着色：LeetCode-colorization-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