요약
ONNX 빠른 R-CNN 모델을 IR로 변환하기 위해 입력 모양을 NCHW로 변경하는 모델 최적화기 명령.
설명
- 명령을 사용하여 더 빠른RCNN-10.onnx 모델을 IR로 변환합니다.
python ./mo_onnx.py --input_model FasterRCNN-10.onnx --input_shape [3,960,960] --mean_values [102.9801,115.9465,122.7717] --transformations_config ./extensions/front/onnx/faster_rcnn.json
- 출력:
[ ERROR ] Exception occurred during running replacer "REPLACEMENT_ID" (): Numbers of inputs and mean/scale values do not match.
해결 방법
- 더 빠른RCNN-10.onnx 모델에는 CHW 입력 모양이 있습니다.
- 추가 --input "0:2" 매개변수 모델 최적화 명령 입력 모양을 NCHW로 변경하려면:
python ./mo_onnx.py --input_model FasterRCNN-10.onnx --input_shape [3,800,800] --mean_values [102.9801,115.9465,122.7717] --transformations_config ./extensions/front/onnx/faster_rcnn.json --input "0:2"