housebw/darknet

housebw/darknet镜像说明

镜像概述

该镜像基于AlexeyAB/darknet项目构建，提供Yolo-v3和Yolo-v2目标检测神经网络的跨平台（Windows/Linux）实现，支持GPU加速（CUDA/CUDNN）和OpenCV集成，可用于实时目标检测、自定义物体检测模型训练等场景。

核心功能

1. 实时目标检测：支持静态图像、视频文件、网络摄像头及手机MJPEG流的目标检测。
2. 自定义模型训练：可基于Pascal VOC数据集或自定义数据集训练专属目标检测模型。
3. 跨平台兼容：适配Windows和Linux系统，满足不同环境需求。
4. 硬件加速：支持CUDA、CUDNN及Tensor Core加速，显著提升检测与训练效率。
5. 多模型支持：包含Yolo-v3、Yolo-v3-tiny、Yolo-v2、Yolo9000等预训练模型，平衡速度与精度。

使用场景

• 静态图像中的物体识别（如图片内行人、车辆检测）。
• 视频文件的目标跟踪与分析（如监控视频中的异常物体识别）。
• 网络摄像头/手机摄像头的实时目标检测（如智能安防场景）。
• 自定义物体检测模型训练（如工业质检、农业病虫害识别等垂直领域需求）。

配置说明

编译选项（Makefile调整）

• GPU=1：启用CUDA加速（需安装CUDA环境）。
• CUDNN=1：启用CUDNN加速（需安装CUDNN库）。
• OPENCV=1：集成OpenCV以支持视频流和摄像头输入。
• LIBSO=1：生成动态链接库（darknet.so）供外部程序调用。

自定义模型训练步骤

1. 修改配置文件（如yolo-obj.cfg）：调整batch、subdivisions参数，更新classes（物体数量）及对应filters（计算公式：(classes+5)*3）。
2. 准备数据集：创建obj.names（物体名称列表）、obj.data（训练配置），标注数据并生成train.txt（训练文件路径列表）。
3. 启动训练：使用预训练权重（如darknet53.conv.74）执行训练命令：
   darknet.exe detector train data/obj.data yolo-obj.cfg darknet53.conv.74

Docker部署示例

静态图像检测

bash
docker run -v $(pwd)/data:/data housebw/darknet detector test /data/coco.data /data/yolov3.cfg /data/yolov3.weights /data/dog.jpg

（注：需将coco.data、yolov3.cfg、yolov3.weights及测试图片dog.jpg放入本地./data目录）

实时摄像头检测

bash
docker run --device=/dev/video0:/dev/video0 -v $(pwd)/data:/data housebw/darknet detector demo /data/coco.data /data/yolov3.cfg /data/yolov3.weights -c 0

（注：需确保容器有权限访问本地摄像头设备/dev/video0）