自动编码器 - 重建缺少颜色

问题描述

我的目标是找到无人监督的全身地标。为此,我使用自动编码器结构来解开全身图像(深度时尚数据集)的形状和外观。损失函数包含一个重建损失,作为副任务需要。

地标检测工作正常,如下图所示。然而,重建图像严重缺乏颜色多样性。

enter image description here

在这张图片上,您可以看到两个输入图像 - 原始图像和图像的增强版本。两个图像都应该重建。您可以按以下顺序查看图像:

输入图像 - 空间表示 - 检测到的关键点 - 重建 - 带有地标的原始图像 - 输入图像增强 - 等等。

如您所见,增强图像的重建完全没有颜色。它根本没有绿色部分,即使损失是相对于增强图像计算的。此外,未增强图像的重建非常“干”。

作为重建损失的损失函数,我在检测到的地标周围使用 L1-Loss。我认为,这可能是次优的。有没有人有想法,如何改进?有没有更适合重建的损失函数?我也尝试过 L2-Loss,但没有改善。

非常感谢!

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)