Media-Smart / vedastr

A scene text recognition toolbox based on PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

What are the correct arguments used in openvino pytorch_to_onnx.py

choozhenbo opened this issue · comments

Since the current GPU is not available to me, the other alternative for me is using CPU. As I run the code, I did encounter these issue. Can you help me to solve it?
python3 /opt/intel/openvino_2021/deployment_tools/tools/model_downloader/src/open_model_zoo/model_tools/internal_scripts/pytorch_to_onnx.py --model-path=/home/choozhenbo/Desktop/convert/public/text-recognition-resnet-fc/ --model-path=/home/choozhenbo/Desktop/convert/text-recognition-resnet-fc --model-name=get_model --import-module=model '--model-param=file_config=r"/home/choozhenbo/Desktop/convert/text-recognition-resnet-fc/vedastr/configs/resnet_fc.py"' '--model-param=weights=r"/home/choozhenbo/Desktop/convert/text-recognition-resnet-fc/vedastr/ckpt/resnet_fc.pth"' --input-shape=1,1,32,100 --input-names=input --output-names=output --output-file=/home/choozhenbo/Desktop/convert/text-recognition-resnet-fc/resnet_fc.onnx
ERROR: Module model contains no class or function with name get_model!
module 'model' has no attribute 'get_model'