將 YOLOv4* 型號轉換為中間表示 (IR) 格式的模型優化指令
- 透過命令將自訂 YOLOv4* 模型轉換為中間表示 (IR):
python mo_tf.py --input_model frozen_darknet_yolov4_model.pb --tensorflow_use_custom_operations_config yolo_v4_tiny.json --批 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 中提供。