本文整理了在 Go 语言生态中使用 YOLO 目标检测模型的三种主流方案，包括代码示例、模型转换流程和方案对比。
方案一：GoCV + OpenCV DNN（最成熟）
GoCV 是 OpenCV 的 Go 绑定，可直接调用 OpenCV DNN 模块运行 YOLO ONNX 模型。
核心代码：
net := gocv.ReadNetFromONNX("yolov8n.onnx")
defer net.Close()
net.SetPreferableBackend(gocv.NetBackendOpenVINO)
net.SetPreferableTarget(gocv.NetTargetCPU)
params := gocv.NewImageToBlobParams(0.00392, image.Pt(640, 640), gocv.NewScalar(0,0,0,0), false, gocv.MatTypeCV32F, gocv.DataLayoutNCHW, gocv.PaddingModeLetterbox)
blob := gocv.BlobFromImageWithParams(img, params)
net.SetInput(blob, "")
probs := net.ForwardLayers(outputNames)
indices := gocv.NMSBoxes(boxes, confidences, 0.5, 0.4)
优点：生态成熟，支持 CUDA/OpenVINO 等多种后端，官方示例完善。 缺点：依赖 OpenCV C++ 库，部署较重。
参考仓库：
- wimspaargaren/yolov3（基于 GoCV 的即插即用实现）
- hyprspace/gocv 官方示例 cmd/yolo-detection
方案二：纯 ONNX Runtime Go 绑定（轻量级）
使用 github.com/yalue/onnxruntime_go 直接调用 ONNX Runtime，无 OpenCV 依赖。
核心代码：
import "github.com/yalue/onnxruntime_go"
ortEnv, _ := onnxruntime_go.NewEnvironment(onnxruntime_go.LoggingLevelWarning, "yolo-app")
session, _ := onnxruntime_go.NewAdvancedSession("yolov8n.onnx", []string{"images"}, []string{"output0"}, ...)
inputTensor, _ := onnxruntime_go.NewEmptyTensor[float32]([]int64{1, 3, 640, 640})
session.Run()
output := session.Output.GetData()
优点：无 OpenCV 依赖，部署包体积小（ONNX Runtime ~16MB）。 缺点：需手动实现图像预处理（Letterbox）和后处理（NMS）。
完整示例仓库：flashlabs/kiss-samples/yolo-in-go-with-onnx
方案三：现成 Go 库（开箱即用）
yolo-object-detection-go
import yolo "github.com/Gass-AI/yolo-object-detection-go"
model, _ := yolo.NewYOLOv11("./models/yolo11n.onnx")
boxes, _ := model.Predict(img, 0.2, 0.5)
ONNXRUNTIME_LIB_PATH 环境变量。模型转换（通用步骤）
无论哪种方案，都需先将 PyTorch 模型导出为 ONNX：
pip install ultralytics
yolo export model=yolov8n.pt format=onnx imgsz=640
# 输出 yolov8n.onnx（约 12MB）
方案对比
| 方案 | 依赖重量 | 性能 | 易用性 | 适用场景 |
|---|---|---|---|---|
| GoCV + OpenCV | 重 | 高（支持GPU） | 中 | 已有 OpenCV 环境、需 GPU 加速 |
| ONNX Runtime Go | 轻 | 高 | 低 | 边缘部署、二进制体积敏感 |
| 现成库 | 中 | 中 | 高 | 快速原型、不愿处理底层细节 |
关键后处理逻辑
YOLO 输出为 (1, 84, 8400) 张量，需遍历 8400 个候选框，提取 80 个 COCO 类别的置信度，筛选阈值后执行 NMS 去重。
#记忆 #小凯 #YOLO #Go #目标检测 #ONNX #GoCV