YOLO检测
功能: 使用深度学习方法对输入图片进行检测
输入参数:
名称 | 类型 | 合法范围 | 缺省值 | 含义 |
---|---|---|---|---|
图像 |
ColorImage |
无 |
无 |
输入图片,需要RGB通道的彩色图片 |
输出参数:
名称 | 类型 | 合法范围 | 缺省值 | 含义 |
---|---|---|---|---|
检测结果 |
DetectInstance |
无 |
{} |
返回包围框,分类,分数和polygon |
初始化: 需要
初始化参数:
名称 | 类型 | 合法范围 | 缺省值 | 含义 |
---|---|---|---|---|
权重文件 |
File |
['.onnx'] |
无 |
模型文件 |
置信阈值 |
Float |
[0.0, 1.0] |
0.8 |
检测阈值 |
开启GPU |
Bool |
无 |
False |
设置是否使用GPU做推理, 若开启需要电脑具有显卡环境,且由于onnxruntime版本与cuda版本有要求, 需要按照官方要求和cuda版本安装cudnn和对应版本onnxruntime-gpu, 见[链接](https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html) |
使用opencv_dnn |
Bool |
无 |
False |
设置是否使用opencv_dnn做推理, 已有测试结果表明此方法打开时检测会稍慢一些 |