dataset=faces;input_height=64;input_width=64;train=true;crop=true;input_fname_pattern=*.jpg
迁移dcgan到ascend910平台上使用NPU运行,并将结果与原论文进行对比
Python 3.6 Tensorflow 1.15 SciPy pillow tqdm
Ascend: 1*Ascend 910 CPU: 24vCPUs 96GiB
DCGAN
└─
├─README.md
├─LICENSE
├─download.py 获取数据集,在NPU上不适用
├─main.py 执行主函数代码
├─model.py 定义模型结构
├─ops.py 自定义算子
├─utils.py 工具函数
数据集:ILSVRC2012_img_train
链接:https://image-net.org/data/ILSVRC/2012/ILSVRC2012_img_train.tar
OBS桶地址:obs://xubinxbchen/imagenet/
python main.py
主要参数注释:
dataset: dataset name
data_dir: dataset url
train: is train
input_height:
output_height: