模型优化器命令,将 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。