用from torchvision.transforms.functional import ssim
之后报错cannot import name 'ssim' from 'torchvision.transforms.functional'
pip install后,仍不行
SSIMLoss目前得通过一个第三方包来使用pytorch_ssim ,这个包目前通过pip安装使用的话会有问题,目前的解决方案是直接把github上的代码下载下来放到项目里面,使用方式很简单。
loss_stru = 1 - pytorch_ssim.ssim(outputs, b_y)
github上的包链接:GitHub - Po-Hsun-Su/pytorch-ssim: pytorch structural similarity (SSIM) loss