Atlas200DK ACL多路推理样例

这个demo主要包括以下功能点

• ffmpeg RTSP/mp4/摄像头视频流输入
• ffmpeg RTMP/RTSP/mp4视频流输出
• DVPP H264解码
• DVPP H264编码(demo中暂时没有开启，因为ACL限制每个进程只能有一个VENC的流，所以现在使用软件编码)
• ACL yolov3推理
• ACL yolov3+deepsort推理

支持版本

20.0以及20.1

环境搭建

环境搭建文档

模型转换

1. 在官方文档的"开发环境"上首先导入atc依赖的环境变量,以这个链接为例。注意install_path要设置为实际安装的路径。

2. 下载yolov3的模型文件

3. 转换模型,yolov3_pp.prototxt和aipp_yolov3_pp.cfg在model文件夹中

atc --model=yolov3_pp.prototxt --weight=yolov3.caffemodel --framework=0 --output=sample-yolov3_pp_416 --soc_version=Ascend310 --insert_op_conf=./aipp_yolov3_pp.cfg

4. 将得到的sample-yolov3_pp_416.om拷贝到model文件夹中

运行测试(RTSP输入RTMP输出)

1. 在Atlas200DK上启动nginx
2. 准备两个rtsp视频流，这里使用ubuntu上的vlc命令行为例。这里的视频文件请替换为你对应的测试文件。(vlc在windows上也可以在节目上完成类似的操作。)

nohup vlc -v road_traffic_test1.mp4 --sout '#rtp{sdp=rtsp://:8554/tt1.mp4}' &
nohup vlc -v road_traffic_test2.mp4 --sout '#rtp{sdp=rtsp://:8555/tt2.mp4}' &

3. 运行run.sh,请将input的地址替换为实际的地址

./run.sh

4. 使用vlc或其他视频播放器检查推理结果。

媒体->打开网络串流->输入run.sh中output对应的rtmp地址。(127.0.0.1替换为Atlas200DK在你的网络中对应的IP)

MP4输入输出

现在新增了对MP4输入输出的支持，可以在run.sh中将rtsp或rtmp的输入输出地址替换为Atlas200 DK本地的MP4文件路径(请尽量使用绝对路径)。

示例:

./build/acl_multi_stream_demo --input input.mp4 --output output.mp4

RTSP输出

现在新增了对RTSP输出的支持，可以在run.sh中将rtmp的输出地址替换为rtsp的路径。

示例：

./build/acl_multi_stream_demo --input input.mp4 --output rtsp://192.168.1.9/stream1

摄像头输入

现在新增了对Atlas200DK摄像头输入的支持，可以在run.sh中将输入地址替换为camera0或camera1。

摄像头输入规格： 720P @ 20fps 示例：

./build/acl_multi_stream_demo --input camera0 --output rtsp://192.168.1.9/stream1

其他格式

由于使用的是ffmpeg通用的API所以其他输入输出格式可能也是天然支持的，你可以直接试一下，说不定可以直接使用呢。

TODO

• 现在测试能够支持2路实时推理，支持更多的推理可能还需要做一些修改(yolov3模型现在是1batch的，效率较低)。
• 现在所有框都是红色的，可以根据不同类别画不同颜色的框