liwang0621

stage 1 的可视化问题

1

我尝试根据Pose2ImagePipeline写了个stage 1的效果可视化，但是效果有点不尽如人意 整体可视化参考的是train_stage_1.py中的log_validation；使用的模型是pretrained_weights文件夹下的denoising_unet.pth、reference_unet.pth、pose_guider.pth；测试数据使用的是configs/inference下的数据，部分测试效果如下：    看这里的可视化效果，整体上感觉很容易出现背景的感觉，大佬有遇到类似的问题吗，还是我使用的模型有误。