johannestang/yolo_service

YOLO目标检测服务镜像

镜像概述

该Docker镜像将基于YOLO的目标检测服务容器化，使用AlexeyAB的darknet分支实现模型推理，并通过connexion框架提供REST API接口，便于快速集成到各类应用场景中。

核心功能

1. 双端点支持：提供detect（返回检测物体类别）和annotate（返回带标注框的图像）两个核心接口；
2. 灵活输入方式：支持GET请求（传入图像URL）和POST请求（上传本地图像文件）；
3. 可视化测试界面：通过localhost:8080/ui可直观测试接口、查看参数详情；
4. 多模型选择：包含YOLOv3（COCO/Open Images数据集）、YOLO9000、YOLOv4（COCO数据集）等多种模型变体；
5. 硬件适配：支持CPU运行，以及基于CUDA 10.0的GPU加速（含Tensor Cores优化版本）。

使用场景

• 图像识别类应用的快速开发与原型验证；
• 监控系统中的物体检测任务；
• 自动化图像分析流程的集成。

配置说明

模型变体标签

• yolov3_coco：基于COCO数据集训练的YOLOv3模型（覆盖80类物体）；
• yolov3_openimages：基于Open Images数据集训练的YOLOv3模型（覆盖601类物体）；
• yolo9000：支持超9000类物体检测的YOLO9000模型；
• yolov4_coco：基于COCO数据集训练的YOLOv4模型（覆盖80类物体）。

Darknet配置标签

• 1.0：CPU运行版本；
• 1.0_cuda10.0：CUDA 10.0加速版本（需GPU硬件支持）；
• 1.0_cuda10.0_tc：CUDA 10.0+Tensor Cores优化版本（需支持Tensor Cores的GPU）。

部署示例

CPU版本部署

bash
docker run -d --rm --name yolo_service -p 8080:8080 johannestang/yolo_service:1.0-yolov3_coco

GPU版本部署（需先安装NVIDIA Container Toolkit）

bash
docker run -d --rm --name yolo_service -p 8080:8080 --gpus all johannestang/yolo_service:1.0_cuda10.0-yolov3_coco