Files
yolov5-inferrence/codes.md
T
2025-08-22 13:52:24 +08:00

1.7 KiB

模型文件说明

1. 说明

efka_flow程序指定的模型目录下(-m参数指定),需要有一下三个文件(下面的模型名称以mymodel为例):

  • mymodel.rknn - 实际模型文件
  • mymodel.txt - 模型类别名称,一行一个名称,分别代表模型识别出的第0类,第1类...的名称
  • mymodel_code.txt - 模型识别之后,上传的事件编码。该编码如果是0,表示不需要上传

2. 实例

这里举了几个例子,用以表示不同情况。

2.1. 工服识别

工服识别的情况,是工服可能有多样性,比如,mymodel.rknn识别出来的0,1,2,3都是不同类型的工服,而第4类,识别出的是普通没穿工服的情况。

此时,我们会希望0,1,2,3类都判定为工服,而4判定为未穿工服。如果有未穿工服的人进入,就报警(假设事件编码是123,表示人员违规进入);如果是正常工服,就生成一条巡检记录(假设事件编码是124,表示正常巡检),则此时mymodel.txt如下:

// mymodel.txt
clothes1
clothes2
clothes3
clothes4
abnormal

mymodel_code.txt应该如下:

// mymodel_code.txt
124
124
124
124
123

如果mymodel_code.txt里面的编码相同,则在上报事件的时候,会被合并。比如视频里面四种类别的工服都有一个,则实际上报的时候,上报的内容如下(这里是举个例子,并不是实际的数据格式):

{
    "class_idx": 124,
    "number": 4,
    // 这里的名称,会合并为第一个出现该编码的类别的名称。
    "name": "clothes1"
}

2.2. 其他例子

待补充。其实,其他例子,均可以由上面的例子推断出来。