1

我想用一个新的图像数据集重新训练一个现有的对象检测模型，并为英特尔 Movidius 量化它。是否有任何工作程序可以做到这一点？

我已经成功地重新训练了模型，但未能量化它。我已按照以下教程重新训练 SSD MobileNet

4

1 回答 1

0

Movidius 设备仅支持 FP16 模型，要转换 SSD Mobilenet 的 caffe 版本，然后将“--data_type FP16”提供给模型优化器 (mo.py) openvino 模型动物园有一个 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]
  • --output=detection_out
  • --input_model=$dl_dir/mobilenet-ssd.caffemodel
  • --input_proto=$dl_dir/mobilenet-ssd.prototxt。

请注意，您的输入形状以及平均值和比例值可能会有所不同，因此请更改它们以匹配您的再训练模型。

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

于 2021-10-19T19:50:11.703 回答