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

Question

我想用新的图像数据集重新训练现有的对象检测模型，并为Intel Movidius量化它。有没有什么工作程序可以做到这一点？

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

Answer 1

Movidius设备只支持FP16模型，要转换咖啡因版本的SSD Mobilenet，然后将"--data_type FP16“提供给模型优化器(mo.py) openvino model zoo有一个mobilenet-ssd模型，也使用caffe，相关的yaml文件具有以下参数model_optimizer_args：

• --input_shape=1,3,300,300
• --input=data
• --mean_values=data127.5,127.5,127.5
• --scale_values=data127.5
• --output=detection_out
• --input_model=$dl_dir/mobilenet-ssd.caffemodel
• --input_proto=$dl_dir/mobilenet-ssd.prototxt.

请注意，您的输入形状以及平均值和缩放值可能会不同，因此请更改这些值以匹配您重新训练的模型。

openvino还附带了一个演示文件，可用于转换后的模型。有关所有支持的体系结构，请参阅相关的mode.lst文件。https://github.com/openvinotoolkit/open_model_zoo/tree/master/demos/object_detection_demo/python