alemelis/darknet

alemelis/darknet镜像说明

镜像概述

本镜像基于NVIDIA CUDA 10.0+CUDNN7环境构建，预编译了支持OpenCV、GPU加速的Darknet框架（AlexeyAB版本），旨在帮助用户快速部署YOLO目标检测模型，避免本地环境配置的繁琐步骤。

核心功能

1. 预集成GPU加速支持，利用CUDA/CUDNN提升目标检测效率
2. 已安装OpenCV、git等依赖，无需手动编译
3. 支持YOLO系列模型（如yolov3-tiny）的快速运行
4. 可通过挂载本地目录使用自定义数据、模型文件和配置

使用场景

• 快速验证YOLO模型的检测效果
• 开发阶段测试目标检测算法
• 避免本地环境中CUDA、OpenCV等依赖的版本冲突问题

配置说明

1. 需安装nvidia-docker以支持GPU加速
2. 挂载本地目录到容器内，确保模型文件、配置文件和测试数据可访问
3. 运行时指定detector test命令及相关参数

部署示例

拉取镜像

bash
docker pull alemelis/darknet:latest

运行YOLO检测（以yolov3-tiny为例）

bash
# 克隆darknet仓库获取配置文件
git clone https://github.com/AlexeyAB/darknet.git
cd darknet
# 下载预训练权重（若未存在）
if [ ! -f yolov3-tiny.weights ]; then
    wget https://pjreddie.com/media/files/yolov3-tiny.weights
fi
# 运行检测
nvidia-docker run -v /home/$(whoami):/home/$(whoami) \
    -w /home/$(whoami)/ --rm --runtime=nvidia alemelis/darknet \
    detector test $(pwd)/cfg/coco.data \
    $(pwd)/cfg/yolov3-tiny.cfg $(pwd)/yolov3-tiny.weights \
    $(pwd)/data/horses.jpg

Dockerfile

dockerfile
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
ARG DEBIAN_FRONTEND=noninteractive
MAINTAINER Alessandro Melis <***>

RUN apt update 
RUN apt install -y \
	pkg-config git build-essential libopencv-dev

# 构建darknet
WORKDIR /app
RUN git clone https://github.com/AlexeyAB/darknet.git
WORKDIR darknet
RUN make OPENCV=1 GPU=1 AVX=1 OPENMP=1 CUDNN=1 CUDNN_HALF=1 OPENMP=1 -j $(nproc)
RUN chmod +x darknet

ENTRYPOINT ["/app/darknet/darknet"]