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