Модель OpenVino вылетает после загрузки сети - PullRequest
Новая
       110

Модель OpenVino вылетает после загрузки сети

0 голосов
/ 11 июля 2020

Я преобразовал модель onnx в IR, используя 

python3 mo_onnx.py --input_model SSD_nms.onnx --input inputs,top_k_per_class,nms_iou,nms_score --input_shape [1,3,224,224],[1],[1],[1]

Этот шаг работал нормально, но когда я пытаюсь загрузить сеть, я получаю следующую ошибку. 

ie = IECore() 
net = ie.read_network(model=model_path, weights=weights_path)
print (net.layers)

File "ie_api.pyx", line 1170, in openvino.inference_engine.ie_api.IENetwork.layers.__get__ ValueError: get_shape was called on a descriptor::Tensor with dynamic shape

Мне удалось успешно запустить ту же модель за вычетом последнего уровня NMS. Не уверен, почему это не удается. Просмотр файла IR xml на netron ничего не показывает.

Полное описание слоя, которое вызывает проблемы: 

<layer id="351" name="selected_indices" type="NonMaxSuppression" version="opset1">
    <data box_encoding="center" sort_result_descending="0"/>
    <input>
        <port id="0">
            <dim>1</dim>
            <dim>10000</dim>
            <dim>4</dim>
        </port>
        <port id="1">
            <dim>1</dim>
            <dim>2</dim>
            <dim>10000</dim>
        </port>
        <port id="2"/>
        <port id="3"/>
        <port id="4"/>
    </input>
    <output>
        <port id="5" precision="I64">
            <dim>10000</dim>
            <dim>3</dim>
        </port>
    </output>
</layer>

Удаление выходного слоя из IR заставляет все работать нормально .

...