Lệnh trình tối ưu hóa mô hình để chuyển đổi mô hình YOLOv4* sang định dạng Đại diện trung gian (IR)
- Chuyển đổi mô hình YOLOv4* tùy chỉnh thành Đại diện trung gian (IR) với lệnh:
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
- Lỗi đã nhận: Không thể suy ra hình dạng hoặc giá trị cho nút "detector/yolo-v4-tiny/Tile/YoloRegion".
Thêm giá trị tỷ lệ và tham số hình dạng đầu vào trong dòng lệnh cho Model Optimizer để chuyển đổi mô hình YOLOv4 thành 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
Việc chuyển đổi mô hình YOLOv4 hiện đã có trong bộ công cụ OpenVINO™ hành 2021.4.