如何重新訓練檢測 model 並為英特爾 Movidius 量化它？

Question

我想用一個新的圖像數據集重新訓練現有的 object 檢測 model 並為英特爾 Movidius 量化它。 有什么工作程序可以做到這一點嗎？

我已經成功地重新訓練了 model 但未能量化它。 我已按照以下教程重新訓練 SSD MobileNet

Answer 1

The Movidius devices only support FP16 models, and to convert a caffe version of SSD Mobilenet then you supply the "--data_type FP16"to the model optimizer (mo.py) The openvino model zoo has a mobilenet-ssd model,also using caffe , 並且關聯的 yaml 文件具有以下參數 model_optimizer_args：

• --input_shape=[1,3,300,300]
• --輸入=數據
• --mean_values=數據[127.5,127.5,127.5]
• --scale_values=數據[127.5]
• --output=detection_out
• --input_model=$dl_dir/mobilenet-ssd.caffemodel
• --input_proto=$dl_dir/mobilenet-ssd.prototxt。

請注意，您的輸入形狀以及平均值和比例值可能會有所不同，因此請更改它們以匹配您重新訓練的 model。

openvino還附帶一個演示文件，可用於轉換后的 model。 請參閱關聯的 mode.lst 文件以了解所有受支持的架構。 https://github.com/openvinotoolkit/open_model_zoo/tree/master/demos/object_detection_demo/python