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 注意事项:
2022-01-12 19:17:54 524KB tensorflow autoencoder rgb-image face-recognition
1