上传者: 42136365
|
上传时间: 2022-01-12 19:17:54
|
文件大小: 524KB
|
文件类型: -
TF_Convolutional_Autoencoder
用于在TensorFlow中以高压缩率对RGB图像进行编码/解码的卷积自动编码器
这是从Arash Saber Tehrani的Deep-Convolutional-AutoEncoder教程教程改编而成的示例模板,用于对3通道图像进行编码/解码。 该模板已被完全注释。 我已经在来自香港中文大学的CelebA数据集的重新缩放后的样本上测试了此实现,以在短时间的训练中产生相当不错的结果。 此实现的压缩比为108。即,对于形状为[-1、48、48、3]的输入张量,瓶颈层已减小为形状为[-1、64]的张量。
附加功能:
将3通道图像而不是MNIST用作输入
培训现在执行检查点保存和还原
可以在TensorBoard中查看编码器的输入和解码器的输出
输入自动重定标
用ReakyReLU代替ReLU激活来解决垂死的ReLU
注意事项: