Pix2Pix深度学习模型生成夜间可见光图像(AI-VIS)开发日志

xiaoqianWX · 发表于 2024-6-23 00:45

补一张在训练的模型目前的预览图
还有很多训练要走，大概过个两天再看看具体是个什么情况

我不是Carl2 · 发表于 2024-7-19 11:02

2024/07/05

之前使用LPIPS的时候错用了“alex”网络作为损失函数，导致训练后期图像出现网格状纹路，SSIM表现也越来越不稳定。目前替换为了“vgg”网络作为损失函数，显存占用更大但收敛更快，训练56epoch后表现如下。目前来看替换预训练网络之后网格状纹路的问题消失了，新模型的高频细节比用SSIM炼的模型更好，但和真实的0.5km vis仍有一定差距。

xiaoqianWX · 发表于 2024-7-19 19:59

我不是Carl2 发表于 2024-7-19 11:02
2024/07/05

之前使用LPIPS的时候错用了“alex”网络作为损失函数，导致训练后期图像出现网格状纹路，SSIM ...

从7月1日开始练到现在，总计练了差不多400小时了
表现好的版本在持续更新到https://ai-vis.dapiya.top/
目前看来这个模型作为正式使用表现相比原模型好的多，虽然SSIM可能没有那么高

tn268261 · 发表于 2024-9-3 23:31

提示: 该帖被管理员或版主屏蔽

		自动登录	找回密码
密码			立即注册

[专业画图] Pix2Pix深度学习模型生成夜间可见光图像(AI-VIS)开发日志

本帖子中包含更多资源

本帖子中包含更多资源