Pytorch的tensor轉圖像需注意的問題


記錄一下自己在實驗中發現的一個問題,我使用了別人的評測函數(matlab寫的),我自己用python實現了一個,通過對生成圖像和圖像標簽進行評測,結果吻合,實現沒問題。

但有趣的是我在訓練過程中,得到的最優模型,在python中得到的指標與matlab對生成結果評測卻不相同。通過控制變量,找到了原因所在,開始我轉圖像是通過tensor轉numpy,然后通過scipy.misc.imsave轉成圖片格式,matlab測的指標有所降低。后來我直接將tensor通過transforms.ToPILImage轉成PIL image格式,然后轉成圖片格式,此時matlab測的指標與python中一樣。

建議不要使用scipy.misc.imsave函數,還是使用torchvision中的transforms,查了一下transforms.ToPILImage,有保護值范圍,scipy.misc.imsave應該是沒有,導致指標下降。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM