Skip to content

yolov5_加载模型

函数简介

Yolov5InitModel必须先加载模型成功才可以调用相关识别函数

函数原型

cpp
LONG Yolov5InitModel(cpu_num, use_gpu, param, param_length, bin, bin_length, names, names_length, password)

参数定义

cpu_num: 识别使用的CPU线程数,范围1-8,使用显卡的情况下该值要设为1

use_gpu: -1代表只使用CPU,0开始代表使用GPU加速(A卡、N卡、集成显卡均可以),如果有多块显卡0代表第一块显卡,1代表第二块显卡

param: 读入param模型文件二进制流数据取变量数据地址

param_length: param的字节集数据长度

bin: 读入bin模型文件二进制流数据取变量数据地址

bin_length: bin的字节集数据长度

names: 读入names文件二进制流数据取变量数据地址

names_length: names的字节集数据长度

password: 若模型有加密则填入密码,否则留空,或者传空字符串

返回值

1: 加载模型成功

0: 加载模型失败,模型已经加载成功了,不能再次加载

-2: 加载模型时发生错误,可能是模型密码错误或模型文件损坏

-3: 解析参数失败,*.param文件损坏

-88: 未注册插件,无法调用本函数

易语言调用例子(来自离火插件网盘 - 易语言图色_yolov5识别例子)


主题切换
 全屏
复制代码

离火插件易语言模块源码


主题切换
 全屏
复制代码

离火AI插件官方网址 https://www.lihuochajian.com