ChromaGAN:ChromaGAN的正式实施 源码

色度 Kerr正式执行ChromaGAN:具有语义类别分布的对抗图片着色[ ] [ ] [ ] 网络架构 先决条件 Linux 的Python 3 NVIDIA GPU + CUDA CuDNN(CPU模式和不带CuDNN的CUDA可能需要很少的修改,但未经测试) 入门 克隆存储库 git clone https://github.com/pvitoria/ChromaGAN cd ChromaGAN/ 要求 pip install -r requirements.txt 下载数据集 下载数据集并将其放在/DATASET/文件夹中。 我们已经使用ImageNet数据集训练了模型。 您可以从下载 网络参数 可以从config.py文件中修改所有参数。 注意:在DATASET的配置文件中修改数据集的名称。 对于每个测试,您都可以在TEST_NAME修改文件夹名称。 变量PRETRAIN