模型库
模型库
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=...),您可以获取所选 zoo 中所有可用模型的列表。- 将
model_path传递给神经网络节点:
Python
1neural_network.setBlobPath(model_path)下载的 blob 的默认路径是
~/.cache/blobconverter 目录。此路径是任意的,可以通过设置 output_dir 参数来指定。BlobConverter 界面
2022.1 及以上可用。有关转换过程的更多信息,请在此处 查找。