smirnou/watsor.gpu

Watsor

镜像概述和主要用途

Watsor 是一款基于深度学习的视频流目标检测工具，主要用于监控场景。它通过实时分析最新帧实现纯粹的实时处理，能对检测到的威胁做出最快反应。

核心功能和特性

• 基于人工神经网络执行智能检测，显著减少视频监控中的误报。
• 支持使用带alpha通道的https://github.com/asmirnou/watsor#zones-and-masks%E6%9D%A5%E9%99%90%E5%88%B6%E6%A3%80%E6%B5%8Bhttps://github.com/asmirnou/watsor#zones-and-masks%E3%80%82
• 支持多种硬件加速器，如Coral USB加速器和Nvidia CUDA GPU，以加速检测算法。
• 通过MQTT协议报告检测到的对象，主要用于与HomeAssistant集成。
• 允许使用通过MQTT发布的命令控制视频解码器。
• 通过HTTP广播带目标检测渲染的MPEG-TS和Motion JPEG格式视频流。
• 从任何源捕获视频，并使用FFmpeg编码带目标检测渲染的任何格式视频。

使用场景和适用范围

除适用于闭路电视（CCTV）外，Watsor 还适用于其他需要视频流目标检测的领域。

使用方法和配置说明

基本运行要求

运行 Watsor Docker 容器时，需将https://github.com/asmirnou/watsor#configuration%E6%96%87%E4%BB%B6%E6%8C%82%E8%BD%BD%E5%88%B0%E5%AE%B9%E5%99%A8%E7%9A%84 /etc/watsor 目录。添加视频和检测硬件加速所需的主机设备（如可用）。由于 Watsor 运行多个共享内存的进程，需将默认64m的共享内存大小增加到更大值，具体取决于摄像头数量。

GPU加速版本使用

要运行 GPU 加速的 watsor.gpu Docker 镜像，需使用https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(Native-GPU-Support)%E3%80%82%E9%80%9A%E8%BF%87 --gpus all 标志将GPU设备添加到容器：

bash
docker run -t -i \
    --rm \
    --env LOG_LEVEL=info \
    --volume /etc/localtime:/etc/localtime:ro \
    --volume $PWD/config:/etc/watsor:ro \
    --device /dev/bus/usb:/dev/bus/usb \
    --device /dev/dri:/dev/dri \
    --publish 8080:8080 \
    --shm-size=512m \
    --gpus all \
    smirnou/watsor.gpu:latest

性能优化

如果您的GPU支持半精度（也称为FP16），可通过以下方式启用此模式以提升性能：

bash
docker run --gpus all --env TRT_FLOAT_PRECISION=16 ...