【图像检索】resnet50由torch转onnx转openvino

0.环境

# yolov5的环境 + onnx
onnx==1.9.0
 
# openvino的环境
openvino_2021.3.394

1.转onnx

torch.onnx.export

2.转openvino

命令参考：open_model_zoo/model.yml at master · openvinotoolkit/open_model_zoo · GitHub

python3 /opt/intel/openvino/deployment_tools/model_optimizer/mo.py --input_model models/resnet50.onnx --output_dir models/resnet50_retrival  --reverse_input_channels --input_shape [1,3,224,224] --mean_values  [0.485,0.456,0.406] --scale_values [0.229,0.224,0.225]  --output Flatten_n

其中，

--mean_values：修改为自己的；

--output Flatten_n：通过netron.app查看。也是你需要输出的对应节点