视频分析环境配置

SuperMap iDesktopX 提供的视频分析、机器学习功能,需要依赖 Python 环境与脚本才能执行,因环境包占用的磁盘空间较大,基础产品包中未提供相关环境,若需使用视频分析、机器学习等功能,需下载扩展包,并进行简单的环境配置。视频分析的环境配置详细说明如下:

推荐配置

  • 处理器:四核 3.7 GHz主频
  • 内存要求:8 GB 或以上
  • 显卡:2GB或以上,独立显卡(安装最新显卡驱动)

若需使用GPU运行环境执行相关分析,计算机硬件配置还需满足以下条件:

  • NVIDIA显卡
  • 显存≥6G
  • 显卡驱动≥411.31

1、下载扩展包

获取SuperMap iDesktopX产品包之后,需下载SuperMap iDesktopX Extension_AI for Windows,建议将其下载至SuperMap iDesktopX产品包中,可在SuperMap 官网技术资源中心下载。

扩展包中提供了以下资源:

  • lib:libtensorflow_jni_gpu-1.15.0.jar;
  • resources_ml:机器学习学习资源包,包含了示范数据、模型配置文件等资源;
  • support
    • MiniConda:机器学习Python的CPU运行环境。
    • video-detection:视频分析所需的分析环境与模型;
  • templates:AR特效资源。

2、获取YOLOv4

SuperMap iDesktopX中的视频分析需要依赖于YOLOv4和扩展包中的Python环境。设置Python环境后,桌面会记录Python环境所在路径,下一次启动桌面可自动启动已配置的Python环境。详细步骤如下:

     2.1 获取YOLOv4资源:在https://gitee.com/SuperMapDesktop/deep-sort-yolov4网站中,根据桌面版本号下载对应版本号的YOLOv4资源包。

     2.2 将下载的deep-sort-yolov4(若下载trunk版本,则名称为:deep-sort-yolov4-trunk,不同版本或下载方式得到的文件夹名称略有不同)文件夹中的内容拷贝到“SuperMap iDesktopX产品包/support/video-detection/deep-sort-yolov4”文件夹中。

3、配置Python环境

     3.1 打开 SuperMap iDesktopX,单击开始->浏览->Python 按钮,在 Python 窗口左侧工具栏处,单击Python环境管理按钮 。

     3.2 在Python环境管理对话框选择“SuperMap iDesktopX产品包/support/MiniConda/ conda / python.exe”环境,单击确定按钮,并重启Python环境。

4、高性能视频分析环境

为了提高视频分析的检测性能,建议配置 Redis 环境。在使用高性能检测前,确保 Redis已启动。Redis 环境配置说明如下:

     4.1 下载 Redis,并将其解压;

     4.2 打开命令行窗口,并进入Redis产品包目录;

     4.3 执行以下命令启动Redis:redis-server.exe redis.windows.conf;

     4.4 修改SuperMap iDesktopX参数:打开iDesktopX产品包/configuration/Desktop.Parameter.xml文件,将highPerformanceDetection="false"改为highPerformanceDetection="true",启动桌面即可。

5、推送直播流

视频分析结果可推送为直播流,便于Web端查看分析结果。但此过程需要进行额外的环境部署,操作说明如下:

     5.1 从官网下载ffmpeg,下载地址:https://github.com/BtbN/FFmpeg-Builds/releases

     5.2 解压后将ffmpeg.exe拷贝到产品包根目录/support/video-detection/Tools/目录下;

6、TensorRT模型转换

iDesktopX支持将YOLOv5 Torch模型转换为TensorRT格式,目前仅windows 10 支持,环境部署说明如下:

     6.1 下载Miniconda3,下载地址:https://docs.conda.io/en/latest/miniconda.html,选择3.8版本,点击下载后,安装即可。

     6.2 载TensorRT包并解压,点击下载以下包地址:https://developer.nvidia.com/nvidia-tensorrt-8x-download

     6.3 在Windows的开始菜单栏中,以管理员身份运行Miniconda3,弹出控制台窗口;

     6.4 输入 conda activate conda路径,conda 路径是AI扩展包解压配置后的路径;

     6.5 控制台进入TensorRT包解压后目录,安装 graphsurgeon、uff、onnx_graphsurgeon、TensorRT,依次执行以下命令:

  • cd .\graphsurgeon\
  • pip install .\graphsurgeon-0.4.6-py2.py3-none-any.whl
  • cd ../
  • cd .\uff\
  • pip install .\uff-0.6.9-py2.py3-none-any.whl
  • cd ../
  • cd .\onnx_graphsurgeon\
  • pip install .\onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl
  • cd ../
  • cd .\python\
  • pip install .\tensorrt-8.6.1-cp38-none-win_amd64.whl

     6.6 将TensorRT包解压后目录lib文件夹中所有dll文件拷贝到产品包根目录/support/video-detection/ TensorRT/下