OpenVino的MXnet模型轉換



[在使用Movidius的模型優化器轉換模型之前,需要先用MXNet的deploy.py將模型轉換成部署模式,然后才能用movidius的優化器轉換]

https://github.com/apache/incubator-mxnet/blob/master/example/ssd/deploy.py

cd ~
git clone https://github.com/apache/incubator-mxnet

mv tmp/*-0000.params tmp/ssd_resnet50_512-0000.params
mv tmp/*-symbol.json tmp/ssd_resnet50_512-symbol.json


python3 incubator-mxnet/example/ssd/deploy.py --network resnet50 --data-shape 512 --num-class 5 --prefix tmp/ssd_

cd /opt/intel/openvino/deployment_tools/model_optimizer

python3 mo_mxnet.py --input_model ssd_resnet50_512-0000.params --input_shape [1,3,512,512]

python3 mo_mxnet.py --input_model ssd_resnet50_512-0000.params --input_shape [1,3,512,512] --data_type=FP16

 

 


免責聲明!

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



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