有人知道如何在英特尔神经计算棒2设备上训练来自detectron2模型库的基于PyTorch的faster_rcnn_R_50_FPN_3x
模型吗?
我已经安装了openvino-toolkit
,并在我的设备上成功运行了其中一个演示。
我已经在google-colab-gpu
上使用detectron2对自定义数据集训练了faster_rcnn_R_50_FPN_3x
模型,现在我需要在英特尔神经计算棒2 VPU上进行训练,以便比较AP结果。
回答:
您无法在英特尔神经计算棒2 (NCS2) 上训练模型。NCS2 应与 OpenVINO™ 工具包一起使用。OpenVINO™ 工具包包括两个主要组件,即模型优化器和推理引擎。推理引擎的 MYRIAD 插件是为在英特尔® 神经计算棒2上进行神经网络推理而开发的。请查看此OpenVINO™ 工具包工作流程。