概要
YOLOv4* モデルを中間表現 (IR) 形式に変換するモデル・オプティマイザーのコマンド
詳細
- コマンドでカスタム YOLOv4* モデルを中間表現 (IR) に変換:
python mo_tf.py --input_model frozen_darknet_yolov4_model.pb --tensorflow_use_custom_operations_config yolo_v4_tiny.json --batch 1 --reverse_input_channels
- 受信エラー:ノード「検出器/yolo-v4-tiny/Tile/YoloRegion」の形状または値を推論できません。
解決方法
モデル・オプティマイザーのコマンドラインでスケール値と入力シェイプ・パラメーターを追加して、YOLOv4 モデルを IR に変換します。
python mo_tf.py --input_model=\Intel\openvino_2021.3.394\deployment_tools\open_model_zoo\tools\downloader\public\yolo-v4-tf\yolo-v4.pb --output_dir="」 --data_type=FP32 --model_name=yolo-v4-tf --input_shape=[1,608,608,3] --input=image_input --scale_values=image_input[255] --reverse_input_channels
追加情報
YOLOv4 モデルの変換は、OpenVINO™・ツールキット 2021.4 で利用できるようになりました。