- YOLACT是一种简单,实时的全卷积分割模型。
运行请注意修改${SOPHON_PIPELINE}/release/yolact_demo/cameras_yolact.json
配置:
{
"cards": [ # 若需要配置多个device,可以在cards下添加多组devid和cameras信息
{
"devid": 0, # 设备id
"cameras": [ # 若需要配置多个视频码流,可以在cameras下添加多组address和chan_num信息。若配置了多个address或多个cards,总的视频码流路数为所有的[chan_num]数量之和
{
"address": "./elevator-1080p-25fps-4000kbps.h264", # 需要测试视频码流的地址,如果是本地文件,只支持h264/h265格式
"chan_num": 1, # 将内容为上述[address]的视频码流配置[chan_num]数量的路数。默认设置为1,会接入1路的内容为上述[address]的视频码流。
"model_names": ["ex1"] # 测试该[address]视频码流的模型名称,需要和此配置文件下面的[models]参数内的模型自定义名称[name]一致,表示使用该模型,多个模型的名字用逗号分开。
}
]
},
],
"pipeline": { # pipeline中的线程数和队列长度
"preprocess": {
"thread_num": 4, # 预处理线程数
"queue_size": 16 # 预处理队列最大长度
},
"inference": {
"thread_num": 1, # 推理线程数
"queue_size": 16 # 推理队列最大长度
},
"postprocess": {
"thread_num": 4, # 后处理线程数
"queue_size": 16 # 后处理队列最大长度
}
},
"models":[
{
"name": "ex1", # 对应于[path]的模型自定义名称
"path": "your_bmodel_path.bmodel", # 对应[name]的bmodel模型的路径
"model_type": "yolact_base", # 所选bmodel的模型类型,需要根据使用的bmodel选择对应的模型类型,否则可能会影响检测精度。支持yolact系列模型(yolact_base、yolact_darknet53、yolact_resnet50和yolact_im700),本例程提供模型类型为:yolact_base、yolact_darknet53和yolact_resnet50。
"skip_frame_num": 0, # 隔帧检测的跳帧数量。当设置为0时表示程序不跳帧检测,当设置为1时表示程序每间隔1帧做一次模型的pipeline。
"output_path": "output_path", # 输出地址,只支持rtsp,tcp 格式为protocol://ip:port/, 例如rtsp://192.168.0.1:8554/test , tcp://172.28.1.1:5353。对于rtsp推流,地址为rtsp server配置的地址。对于tcp,需要开放自己配置的端口。
"obj_threshold": 0.5, # 对应[path]的bmodel模型后处理的物体置信度阈值
"nms_threshold": 0.5, # 对应[path]的bmodel模型后处理的非极大值抑制阈值
}
]
}
NOTE
线程数和队列长度可根据设备情况自行定义。原则上,预处理线程数和后处理线程数设置为设备的逻辑CPU的个数。推理线程数单个pipeline一般为1。
NOTE
测试视频下载:
python3 -m dfss [email protected]:sophon-pipeline/common/elevator-1080p-25fps-4000kbps.h264
模型下载:
python3 -m dfss [email protected]:sophon-pipeline/models/yolact.tar.gz
参数说明
Usage: yolact_demo [params]
--config (value:./cameras_yolact.json)
cameras_yolact.json配置文件的路径,默认路径为./cameras_yolact.json。
--help (value:true)
打印帮助信息
以设置cameras_yolact.json
的chan_num=1
为例测试示例如下:
cd ${SOPHON_PIPELINE}/release/yolact_demo
# ./x86/yolact_demo --help 查看命令行帮助信息
# 以x86 pcie 1684x yolact_base模型为例,将下载好的yolact_base模型拷贝到${SOPHON_PIPELINE}/release/yolact_demo目录下运行
./x86/yolact_demo --config=./cameras_yolact.json
执行会打印如下信息:
# 以x86 pcie 1684x yolact_base模型为例
# 先打印出每路(1路)视频码流及对应芯片相关信息,再打印1路检测器det的总FPS和第0路视频码流处理对应的speed信息。其中,FPS和speed信息与当前运行设备的硬件配置相关,不同设备运行结果不同属正常现象,且同一设备运行程序过程中FPS和speed信息有一定波动属于正常现象。FPS和speed信息如下所示:
...
[2023-04-22:16:01:21] total fps =-nan,ch=0: speed=-nan
[2023-04-22:16:01:22] total fps =24.0,ch=0: speed=24.0
[2023-04-22:16:01:23] total fps =25.3,ch=0: speed=25.3
[2023-04-22:16:01:24] total fps =25.0,ch=0: speed=25.0
[2023-04-22:16:01:25] total fps =25.0,ch=0: speed=25.0
[2023-04-22:16:01:26] total fps =25.0,ch=0: speed=25.0
[2023-04-22:16:01:27] total fps =25.0,ch=0: speed=25.0
[2023-04-22:16:01:28] total fps =25.0,ch=0: speed=25.0
...
将交叉编译好的${SOPHON_PIPELINE}/release/yolact_demo
文件夹下的cameras_yolact.json
、soc
文件夹以及对应的模型、测试视频一起拷贝到arm SoC运行设备的同一目录下,并修改好cameras_yolact.json的相应配置,运行:
cd ${SOPHON_PIPELINE_YOLACT}
# ./soc/yolact_demo --help 查看命令行帮助信息
# 以arm SoC 1684x yolact_base模型为例
./soc/yolact_demo --config=./cameras_yolact.json
执行会打印如下信息:
# 以arm SoC 1684x yolact_base模型为例
# 先打印出每路(1路)视频码流及对应芯片相关信息,再打印1路检测器det的总FPS和第0路视频码流处理对应的speed信息。其中,FPS和speed信息与当前运行设备的硬件配置相关,不同设备运行结果不同属正常现象,且同一设备运行程序过程中FPS和speed信息有一定波动属于正常现象。FPS和speed信息如下所示:
...
[2023-04-22:16:00:26] total fps =nan,ch=0: speed=nan
[2023-04-22:16:00:27] total fps =24.0,ch=0: speed=24.0
[2023-04-22:16:00:28] total fps =24.0,ch=0: speed=24.0
[2023-04-22:16:00:29] total fps =25.4,ch=0: speed=25.4
[2023-04-22:16:00:30] total fps =25.0,ch=0: speed=25.0
[2023-04-22:16:00:31] total fps =25.0,ch=0: speed=25.0
[2023-04-22:16:00:32] total fps =25.0,ch=0: speed=25.0
[2023-04-22:16:00:33] total fps =25.0,ch=0: speed=25.0
...
以设置cameras_yolact.json
的chan_num=1
为例测试示例如下:
cd ${SOPHON_PIPELINE}/release/yolact_demo
# ./arm64/yolact_demo --help 查看命令行帮助信息
# 以arm pcie 1684x yolact_base模型为例,将下载好的yolact_base模型拷贝到${SOPHON_PIPELINE}/release/yolact_demo目录下运行
./arm64/yolact_demo --config=./cameras_yolact.json
执行会打印如下信息:
# 以arm pcie 1684x yolact_base模型为例
# 先打印出每路(1路)视频码流及对应芯片相关信息,再打印1路检测器det的总FPS和第0路视频码流处理对应的speed信息。其中,FPS和speed信息与当前运行设备的硬件配置相关,不同设备运行结果不同属正常现象,且同一设备运行程序过程中FPS和speed信息有一定波动属于正常现象。FPS和speed信息如下所示:
...
[2023-04-22:18:01:21] total fps =-nan,ch=0: speed=-nan
[2023-04-22:18:01:22] total fps =24.0,ch=0: speed=24.0
[2023-04-22:18:01:23] total fps =25.3,ch=0: speed=25.3
[2023-04-22:18:01:24] total fps =25.0,ch=0: speed=25.0
[2023-04-22:18:01:25] total fps =25.0,ch=0: speed=25.0
[2023-04-22:18:01:26] total fps =25.0,ch=0: speed=25.0
[2023-04-22:18:01:27] total fps =25.0,ch=0: speed=25.0
[2023-04-22:18:01:28] total fps =25.0,ch=0: speed=25.0
...
- 使用pipeline_client显示实时流和检测结果