문서 ID: 000058700 콘텐츠 형태: 문제 해결 마지막 검토일: 2022-05-20

사용자 지정 교육을 받은 SSD MobileNetV2를 IR(중간 표현)으로 변환할 때 실행 중인 교체기 "REPLACEMENT_ID"에서 오류가 발생했습니다.

환경

오픈 소스 OpenVINO™ 툴킷 2021.2  TensorFlow* 2.2 TensorFlow 1.15

BUILT IN - ARTICLE INTRO SECOND COMPONENT
요약

모델 최적화기를 사용하여 사용자 정의 SSD MobileNet V2 모델을 IR로 변환하는 단계

설명

사전 교육을 받은 SSD MobileNetV2 모델을 IR로 변환했지만 맞춤 교육을 받은 모델을 변환할 수 없습니다.

  1. 내보낸 냉동 모델 그래프:

    python object_detection/export_inference_graph.py \ --input_type=image_tensor \ --pipeline_config_path={PIPELINE_CONFIG_PATH} \ --output_directory="exported_model" \ --trained_checkpoint_prefix="/content/model/research/helmet_detector/model.ckpt-10000"

  2. Model Optimizer를 사용하여 냉동 모델 그래프를 IR로 변환하려고 시도했습니다.

    python mo_tf.py \ --input_model ./exported_model/frozen_inference_graph.pb \ --tensorflow_use_custom_operations_config 확장/전면/tf/ssd_v2_support.json \ --tensorflow_object_detection_api_pipeline_config ./helment_detector_tf1.config \ --input_shape [1,300,300,3] \ --reverse_input_channels \ --output_dir output_ncs \ --data_type FP16

  3. 오류가 발생했습니다.

    [ 오류] 교체기 "REPLACEMENT_ID"(): 노드 StatefulPartitionedCall/Postprocessor/BatchMultiClassNonMaxSuppression/map/while에 대한 속성을 추출하는 동안 예기치 않은 예외가 발생했습니다. 원래 예외 메시지: '^Postprocessor/BatchMultiClassNonMaxSuppression/map/while/MultiClassNonMaxSuppression/SortByField/Assert/Assert'

해결 방법

모델 최적화기 변환 명령에서 TensorFlow* 1 모델( ssd_support_api_v1.15.json)으로 생성된 SSD MobileNetV2 모델에 적합한 사용자 지정 작업 구성 파일을 사용하십시오. 변환 지침은 다음 페이지를 참조하십시오 . 모델 변환 방법

python3 ./openvino/model-optimizer/mo_tf.py --input_model ./detector/exported_model/frozen_inference_graph.pb --tensorflow_use_custom_operations_config ./openvino/model-optimizer/extensions/front/tf/ssd_support_api_v1.15.json --tensorflow_object_detection_api_pipeline_config ./detector/helment_detector_tf1.config --input_shape [1,300,300,3] --reverse_input_channels --data_type FP16

관련 제품

이 문서는 다음 항목에 적용됩니다. 2 제품

이 페이지의 콘텐츠는 원본 영어 콘텐츠에 대한 사람 번역 및 컴퓨터 번역의 조합으로 완성되었습니다. 이 콘텐츠는 편의와 일반적인 정보 제공을 위해서만 제공되었으며, 완전하거나 정확한 것으로 간주되어선 안 됩니다. 이 페이지의 영어 버전과 번역 간 모순이 있는 경우, 영어 버전이 우선적으로 적용됩니다. 이 페이지의 영어 버전을 확인하십시오.