【文件属性】：
文件名称：ChromaGAN:ChromaGAN的正式实施
文件大小：11.6MB
文件格式：ZIP
更新时间：2021-03-09 05:25:26
computer-vision deep-learning adversarial-networks coloring colorization
色度
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
【文件预览】：
ChromaGAN-master
----SAMPLE_IMGS()
--------ILSVRC2012_val_00000023.JPEG(46KB)
----SOURCE()
--------config.py(567B)
--------dataClass.py(2KB)
--------ChromaGAN.py(14KB)
--------ChromaGANPrint.py(5KB)
----DATASET()
--------DummyFile(10B)
----requirements.txt(49B)
----Figures()
--------Results.png(2.72MB)
--------22.09770966_ILSVRC2012_val_00001042._reconstructed.jpg(117KB)
--------10.23894882_ILSVRC2012_val_00000680._reconstructed.jpg(171KB)
--------27.31452560_ILSVRC2012_val_00000186._reconstructed.jpg(350KB)
--------26.34756279_ILSVRC2012_val_00000796._reconstructed.jpg(136KB)
--------8.27914047_ILSVRC2012_val_00000915._reconstructed.jpg(174KB)
--------ColorizationModel.png(882KB)
--------31.34938049_ILSVRC2012_val_00000211._reconstructed.jpg(101KB)
--------23.27637291_ILSVRC2012_val_00000023._reconstructed.jpg(106KB)
--------27.84384155_ILSVRC2012_val_00000013._reconstructed.jpg(131KB)
--------28.42907333_ILSVRC2012_val_00001398._reconstructed.jpg(126KB)
--------28.33658218_ILSVRC2012_val_00049397._reconstructed.jpg(195KB)
----LICENCE(34KB)
----README.md(5KB)
----DemoChromaGAN.ipynb(8.52MB)