有些冷門算法真的不要隨便pip,有可能下載到史前版本…最好還是找源代碼 汗
今天要用到SSIM損失函數,從網上簡單看了一下原理就想測試一下,偷了一下懶就直接在命令行輸入pip install pytorch_ssim了,結果報了一堆錯誤(汗),正如這篇博文所說博文在此,傳入Conv2D中的padding是浮點數,然而這樣做完之后還會報錯,差不多是說img和window不在一個運行設備上,我這邊是window張量沒有在gpu上,結果我用了.cuda()之后還是不管用,然后我就去github上看了一下,才發現這個pip包的版本太落后了。于是我重新下載覆蓋之后,成功運行了!感謝!
點我github源碼
后話:如果你想使用github上給出的案例,請你仿照知乎上這位給出的改進版本,源代碼會報錯。點這里