요약
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=\인텔\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에서 사용할 수 있습니다.