在實際項目中出現的該問題,起初以為是cudnn版本的問題,后來才定位到在網絡進行reshape操作的時候
input_layer->Reshape({(int)imgin.size(), input_layer->shape(1), input_layer->shape(2), input_layer->shape(3)});
如上所示,第一個參數是輸入圖片的尺寸,在實際的視頻中,輸入的圖片尺寸有可能為0,那么在reshape操作的時候就會報錯。
在外層加一個保護就好了。
在實際項目中出現的該問題,起初以為是cudnn版本的問題,后來才定位到在網絡進行reshape操作的時候
input_layer->Reshape({(int)imgin.size(), input_layer->shape(1), input_layer->shape(2), input_layer->shape(3)});
如上所示,第一個參數是輸入圖片的尺寸,在實際的視頻中,輸入的圖片尺寸有可能為0,那么在reshape操作的時候就會報錯。
在外層加一個保護就好了。
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。