chunbaobao / Deep-JSCC-PyTorch

A implement of Deep JSCC for wireless image transmission by PyTorch
64 stars 9 forks source link

关于训练高分辨率图像数据集 #9

Closed xiangyanya closed 4 months ago

xiangyanya commented 4 months ago

你好,我用您的模型训练高分辨率图像数据集DIV2K,但是发现测试的时候,psnr不随snr的变高而变好,反而数值相似,请问你遇到这个问题了吗

chunbaobao commented 4 months ago

没有在这个数据集上做过实验,不过我猜测可能对于高分辨率图像需要使用更高的传输功率P(默认为1)或者更高精度的浮点数以防止计算float数时归零

xiangyanya commented 4 months ago

谢谢你,是有用的