lp6m / yolov5s_android

Run yolov5s on Android device!
GNU General Public License v3.0
311 stars 77 forks source link

no node named Conv_253 #22

Closed yspisgod closed 2 years ago

yspisgod commented 2 years ago

when i run python3 /opt/intel/openvino_2021.3.394/deployment_tools/model_optimizer/mo.py --input_model yolov5s.onnx --input_shape [1,3,640,640] --output_dir ./openvino --data_type FP32 --output Conv_253,Conv_302,Conv_351

it seems that "-output" is unnecessary

lp6m commented 2 years ago

Output node names are different in different environment, and yolov5 version. Check the output node name in Netron.

We remove the yolov5 head layers through model conversion, so '--output' is mandatory.