【文件属性】:
文件名称:VAE-Mnist-CoordConv:适用于MNIST的变分自动编码器的CoordConv(卷积和反卷积)的实现
文件大小:5KB
文件格式:ZIP
更新时间:2021-05-01 01:06:44
Python
适用于MNIST数据集的变分自动编码器的CoordConv(卷积和反卷积)的实现
1.安装依赖项:
pip install -r requirements.txt
2.在没有coordConv图层的情况下进行训练
python trainVAE.py --useCoordConv False
3.使用coordConv图层进行训练
python trainVAE.py --useCoordConv True
4.在TensorBoard上显示重构
tensorboard --logdir logdir
5.比较重建:
python compare_generated_images.py
【文件预览】:
VAE-Mnist-CoordConv-master
----.gitignore(43B)
----requirements.txt(46B)
----network.py(4KB)
----README.md(656B)
----compare_generated_images.py(2KB)
----trainVAE.py(2KB)
----coordConv.py(3KB)