nnie模型(海思平台)平台转模型改变输入消除ceil_mode影响

原因:海思平台不支持maxpool中ceil_mode=false,

resnet中的maxpool默认为false。

https://pytorch.org/docs/stable/nn.html

nnie模型(海思平台)平台转模型改变输入消除ceil_mode影响_第1张图片
 

解决办法:

resnet  maxpool只有一处:

nnie模型(海思平台)平台转模型改变输入消除ceil_mode影响_第2张图片

改变输入大小,使得不受ceil_mode影响。

pool计算公式:

如:输入为384, 卷积后为192

(192+2*1+1*(3-1)-1)/2 +1 = 95.5 + 1,torch 95+1 = 96, nnie:96+1=97

所以改变输入使得相除为整数即可。(191+2*1+1*(3-1)-1)/2 +1 = 95+ 1 = 96,所以使得输入为191,反推输入为382.

 

同理:

输入64->62

输入96-> 94

输入192-> 190                (96+2-3)/2 + 1-->(95+2-3)/2 +1

输入224->222

 

总结使得相应输入减2即可。减2后在降采样后变为减1,消除0.5的ceil_mode影响。

你可能感兴趣的:(车牌识别)