InvalidArgumentError: ConcatOp : Dimensions of inputs should match: shape[0] = [1,136,240,64] vs. shape[1] = [1,135,240,64]


初始輸入圖片大小為1600*1200,設置的輸入網絡的最大測試圖片大小為1600*1200相當於scale =1 ,運行沒有問題。之后輸入圖片大小為1920*1080,測試圖片大小為1920*1080,運行報錯:InvalidArgumentError: ConcatOp : Dimensions of inputs should match: shape[0] = [1,136,240,64] vs. shape[1] = [1,135,240,64]

更改最大測試圖片大小為1600*960,運行成功.

初步分析可能1920*1080 進一步scale后得到270*480,在same pading 和特定的卷積參數下會出現問題,270除以2后得到奇數,進一步卷積可能遇到除法向上取整的問題,盡量將輸入網絡的最大測試圖片取為8,16 36,64這樣的倍數。

 

 


免責聲明!

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



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