這是yolov3-darknet模型轉onnx中遇到的,是由於onnx的不同版本的接口不同導致的,
常用的解決方案是更換onnx的版本,一般啥1.2.1, 1.4.1, 1.5.0根據自己的需要更換:
更換過程如下,假設onnx裝在python2中:
(1) python2 -m pip uninstall onnx
注意:我使用 pip uninstall onnx卸載后,發現進入python2中,onnx的版本並沒有發生改變,建議用(1)卸載
(2) python2 -m pip install onnx==1.5.0
如果在上述更換版本中很麻煩,你可以使用如下辦法:
upsample_node = helper.make_node( 'Upsample', mode='nearest', inputs=inputs, outputs=[layer_name], name=layer_name, ) 改為 upsample_node = helper.make_node( 'Resize', #Upsample -->Resize mode='nearest', inputs=inputs, outputs=[layer_name], name=layer_name, )
