模型库
model.yml 文件。使用方法
请注意,DepthAI 模型库中的模型导出时,期望图像值范围为
[0-255],颜色顺序为 BGR(蓝绿红),通道布局为 CHW(通道高度宽度)。Python(DepthAI)
- 使用
pip安装 BlobConverter:
Command Line
1pip install blobconverter- 调用
from_zoo方法,设置zoo_type="depthai"、所需的模型name以及要使用的shaves数量:
Python
1import blobconverter
2model_path = blobconverter.from_zoo(
3 name="yolop_320x320",
4 zoo_type="depthai",
5 shaves=6
6 )zoo_type="intel" 从 Open Model Zoo 下载模型。 调用 blobconverter.zoo_list(zoo_type=...) 可以获取所选模型库中所有可用模型的列表。- 将
model_path传递给神经网络节点:
Python
1neural_network.setBlobPath(model_path)下载的 blob 文件默认路径为
~/.cache/blobconverter 目录。该路径并非固定,可通过设置 output_dir 参数来指定。BlobConverter 界面
2022.1 及以上可用。您可以在 此处 找到有关转换过程的更多信息。