如何为 Intel Movidius 重新训练检测模型并对其进行量化?

How to retrain a detection model and quantize it for Intel Movidius?

我想使用新的图像数据集重新训练现有的对象检测模型,并针对英特尔 Movidius 对其进行量化。有什么工作程序可以做到这一点吗?

我已经成功地重新训练了模型,但未能对其进行量化。我遵循了以下教程 Retrain SSD MobileNet

Movidius 设备仅支持 FP16 模型,要转换 SSD Mobilenet 的 caffe 版本,您需要将“--data_type FP16”提供给模型优化器 (mo.py) openvino模型zoo有一个mobilenet-ssd模型,同样使用caffe,关联的yaml文件有如下参数 model_optimizer_args:

  • --input_shape=[1,3,300,300]
  • --输入=数据
  • --mean_values=数据[127.5,127.5,127.5]
  • --scale_values=数据[127.5]
  • --输出=detection_out
  • --input_model=$dl_dir/mobilenet-ssd.caffemodel
  • --input_proto=$dl_dir/mobilenet-ssd.prototxt.

请注意,您的输入形状以及平均值和尺度值可能会有所不同,因此请更改它们以匹配您重新训练的模型。

还有一个演示文件随 that can be used with your converted model. See the associated mode.lst file for all the supported architectures. https://github.com/openvinotoolkit/open_model_zoo/tree/master/demos/object_detection_demo/python

一起提供