pytorch訓練模型的一些坑


1. 圖像讀取

opencv的python和c++讀取的圖像結果不一致,是因為python和c++采用的opencv版本不一樣,從而使用的解碼庫不同,導致讀取的結果不同。

詳細內容參考:https://www.cnblogs.com/haiyang21/p/11655404.html

2. 圖像變換

PIL和pytorch的圖像resize操作,與opencv的resize結果不一樣,這樣會導致訓練采用PIL,預測時采用opencv,結果差別很大,尤其是在檢測和分割任務中比較明顯。

3. 數值計算

pytorch的torch.exp與c++的exp計算,10e-6的數值時候會有10e-3的誤差,對於高精度計算需要特別注意,比如

兩個輸入5.6015975.601601, 經過exp計算后變成270.85862343143174270.85970686809225

 


免責聲明!

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



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