在實際項目中出現的該問題,起初以為是cudnn版本的問題,后來才定位到在網絡進行reshape操作的時候
input_layer->Reshape({(int)imgin.size(), input_layer->shape(1),
input_layer->shape(2), input_layer->shape(3)});
如上所示,第一個參數是輸入圖片的尺寸,在實際的視頻中,輸入的圖片尺寸有可能為0,那么在reshape操作的時候就會報錯。
在外層加一個保護就好了。
