:sparkles: support yolo26

README.md

@@ -19,12 +19,10 @@ tags:
    <a href="https://github.com/getcharzp/go-vision/pulls" target="blank">
       <img src="https://img.shields.io/github/issues-pr/getcharzp/go-vision?style=for-the-badge" alt="go-vision pull-requests"/>
    </a>
-   <a href='https://github.com/getcharzp/go-vision/releases'>
-      <img src='https://img.shields.io/github/release/getcharzp/go-vision?&label=Latest&style=for-the-badge'>
-   </a>
 </p>
 
-go-vision 基于 Golang + [ONNX](https://github.com/microsoft/onnxruntime/releases/tag/v1.23.2) 构建的轻量级视觉库
+go-vision 基于 Golang + [ONNX](https://github.com/microsoft/onnxruntime/releases/tag/v1.23.2) 构建的视觉库，支持 SAM2、YOLOv11-Det、YOLOv11-Seg、YOLOv11-Cls、YOLOv11-Pose、YOLOv11-OBB、YOLO26-Det、
+YOLO26-Seg、YOLO26-Cls、YOLO26-Pose、YOLO26-OBB 等模型。
 
 ## 安装
 
@@ -274,3 +272,212 @@ func main() {
 | 原图                                                  | OBB图                                                       |
 |-----------------------------------------------------|------------------------------------------------------------|
 | <img width="100%" src="./examples/ship.jpg" alt=""> | <img width="100%" src="./examples/yolov11_obb.jpg" alt=""> |
+
+
+### yolo26-det
+
+```go
+package main
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolo26"
+	"github.com/up-zero/gotool/imageutil"
+	"image"
+	"image/color"
+	"image/draw"
+	"log"
+)
+
+func main() {
+	engine, err := yolo26.NewDetEngine(yolo26.DefaultDetConfig())
+	if err != nil {
+		log.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./test.png")
+	results, err := engine.Predict(img)
+	if err != nil {
+		log.Fatalf("预测失败: %v", err)
+	}
+
+	targetImg := image.NewRGBA(img.Bounds())
+	draw.Draw(targetImg, img.Bounds(), img, img.Bounds().Min, draw.Src)
+	fmt.Printf("检测到目标: %d 个\n", len(results))
+	for _, res := range results {
+		fmt.Printf("Class: %d, Score: %.2f, Box: %v\n", res.ClassID, res.Score, res.Box)
+		imageutil.DrawThickRectOutline(targetImg, res.Box, color.RGBA{R: 255, G: 0, B: 0, A: 255}, 3)
+	}
+	imageutil.Save("yolo26_det.jpg", targetImg, 50)
+}
+```
+
+| 原图                                                  | 检测结果                                                      |
+|-----------------------------------------------------|-----------------------------------------------------------|
+| <img width="100%" src="./examples/test.png" alt=""> | <img width="100%" src="./examples/yolo26_det.jpg" alt=""> |
+
+### yolo26-seg
+
+```go
+package main
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolo26"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+)
+
+func main() {
+	cfg := yolo26.DefaultSegConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26s-seg.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewSegEngine(cfg)
+	if err != nil {
+		log.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./test.png")
+	results, err := engine.Predict(img)
+	if err != nil {
+		log.Fatalf("预测失败: %v", err)
+	}
+
+	fmt.Printf("检测到目标: %d 个\n", len(results))
+	for idx, res := range results {
+		fmt.Printf("Class: %d, Score: %.2f, Box: %v\n", res.ClassID, res.Score, res.Box)
+		imageutil.Save(fmt.Sprintf("yolo26_seg_mask_%d.png", idx), res.Mask, 100)
+	}
+}
+```
+
+| 原图                                                  | Mask图                                                            |
+|-----------------------------------------------------|------------------------------------------------------------------|
+| <img width="100%" src="./examples/test.png" alt=""> | <img width="100%" src="./examples/yolo26_seg_mask_0.png" alt=""> |
+
+
+### yolo26-cls
+
+```go
+package main
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolo26"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+)
+
+func main() {
+	cfg := yolo26.DefaultClsConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26m-cls.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewClsEngine(cfg)
+	if err != nil {
+		log.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./test.png")
+	results, err := engine.Predict(img, 5)
+	if err != nil {
+		log.Fatalf("预测失败: %v", err)
+	}
+
+	for _, res := range results {
+		fmt.Printf("Class: %d, Score: %.5f\n", res.ClassID, res.Score)
+	}
+}
+
+// 输出：
+//  Class: 656, Score: 0.99169
+//  Class: 436, Score: 0.00258
+//  Class: 511, Score: 0.00242
+//  Class: 654, Score: 0.00054
+//  Class: 675, Score: 0.00053
+```
+
+### yolo26-pose
+
+```go
+package main
+
+import (
+	"github.com/getcharzp/go-vision/yolo26"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+)
+
+func main() {
+	cfg := yolo26.DefaultPoseConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26m-pose.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewPoseEngine(cfg)
+	if err != nil {
+		log.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./person.jpg")
+	results, err := engine.Predict(img)
+	if err != nil {
+		log.Fatalf("预测失败: %v", err)
+	}
+
+	dst := yolo26.DrawPoseResult(img, results)
+	imageutil.Save("yolo26_pose.jpg", dst, 50)
+}
+```
+
+| 原图                                                    | 姿态图                                                        |
+|-------------------------------------------------------|------------------------------------------------------------|
+| <img width="100%" src="./examples/person.jpg" alt=""> | <img width="100%" src="./examples/yolo26_pose.jpg" alt=""> |
+
+### yolo26-obb
+
+```go
+package main
+
+import (
+	"github.com/getcharzp/go-vision/yolo26"
+	"github.com/up-zero/gotool/imageutil"
+	"image"
+	"image/color"
+	"image/draw"
+	"log"
+)
+
+func main() {
+	cfg := yolo26.DefaultOBBConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26m-obb.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewOBBEngine(cfg)
+	if err != nil {
+		log.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./ship.jpg")
+	results, err := engine.Predict(img)
+	if err != nil {
+		log.Fatalf("预测失败: %v", err)
+	}
+
+	dst := image.NewRGBA(img.Bounds())
+	draw.Draw(dst, img.Bounds(), img, img.Bounds().Min, draw.Src)
+	for _, result := range results {
+		imageutil.DrawThickPolygonOutline(dst, result.Corners[:], 3, color.RGBA{R: 255, G: 0, B: 0, A: 255})
+	}
+	imageutil.Save("yolo26_obb.jpg", dst, 50)
+}
+```
+
+| 原图                                                  | OBB图                                                      |
+|-----------------------------------------------------|-----------------------------------------------------------|
+| <img width="100%" src="./examples/ship.jpg" alt=""> | <img width="100%" src="./examples/yolo26_obb.jpg" alt=""> |

examples/yolo26_test.go

@@ -0,0 +1,130 @@
+package examples
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolo26"
+	"github.com/up-zero/gotool/imageutil"
+	"image"
+	"image/color"
+	"image/draw"
+	"testing"
+)
+
+func TestYOLO26Det(t *testing.T) {
+	cfg := yolo26.DefaultDetConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26m.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewDetEngine(cfg)
+	if err != nil {
+		t.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./test.png")
+	results, err := engine.Predict(img)
+	if err != nil {
+		t.Fatalf("预测失败: %v", err)
+	}
+
+	targetImg := image.NewRGBA(img.Bounds())
+	draw.Draw(targetImg, img.Bounds(), img, img.Bounds().Min, draw.Src)
+	fmt.Printf("检测到目标: %d 个\n", len(results))
+	for _, res := range results {
+		fmt.Printf("Class: %d, Score: %.2f, Box: %v\n", res.ClassID, res.Score, res.Box)
+		imageutil.DrawThickRectOutline(targetImg, res.Box, color.RGBA{R: 255, G: 0, B: 0, A: 255}, 3)
+	}
+	imageutil.Save("yolo26_det.jpg", targetImg, 50)
+}
+
+func TestYOLO26Seg(t *testing.T) {
+	cfg := yolo26.DefaultSegConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26s-seg.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewSegEngine(cfg)
+	if err != nil {
+		t.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./test.png")
+	results, err := engine.Predict(img)
+	if err != nil {
+		t.Fatalf("预测失败: %v", err)
+	}
+
+	fmt.Printf("检测到目标: %d 个\n", len(results))
+	for idx, res := range results {
+		fmt.Printf("Class: %d, Score: %.2f, Box: %v\n", res.ClassID, res.Score, res.Box)
+		imageutil.Save(fmt.Sprintf("yolo26_seg_mask_%d.png", idx), res.Mask, 100)
+	}
+}
+
+func TestYOLO26Cls(t *testing.T) {
+	cfg := yolo26.DefaultClsConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26m-cls.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewClsEngine(cfg)
+	if err != nil {
+		t.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./test.png")
+	results, err := engine.Predict(img, 5)
+	if err != nil {
+		t.Fatalf("预测失败: %v", err)
+	}
+
+	for _, res := range results {
+		fmt.Printf("Class: %d, Score: %.5f\n", res.ClassID, res.Score)
+	}
+}
+
+func TestYOLO26Pose(t *testing.T) {
+	cfg := yolo26.DefaultPoseConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26m-pose.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewPoseEngine(cfg)
+	if err != nil {
+		t.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./person.jpg")
+	results, err := engine.Predict(img)
+	if err != nil {
+		t.Fatalf("预测失败: %v", err)
+	}
+
+	dst := yolo26.DrawPoseResult(img, results)
+	imageutil.Save("yolo26_pose.jpg", dst, 50)
+}
+
+func TestYOLO26OBB(t *testing.T) {
+	cfg := yolo26.DefaultOBBConfig()
+	cfg.ModelPath = "../yolo26_weights/yolo26m-obb.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolo26.NewOBBEngine(cfg)
+	if err != nil {
+		t.Fatalf("初始化引擎失败: %v", err)
+	}
+	defer engine.Destroy()
+
+	img, _ := imageutil.Open("./ship.jpg")
+	results, err := engine.Predict(img)
+	if err != nil {
+		t.Fatalf("预测失败: %v", err)
+	}
+
+	dst := image.NewRGBA(img.Bounds())
+	draw.Draw(dst, img.Bounds(), img, img.Bounds().Min, draw.Src)
+	for _, result := range results {
+		imageutil.DrawThickPolygonOutline(dst, result.Corners[:], 3, color.RGBA{R: 255, G: 0, B: 0, A: 255})
+	}
+	imageutil.Save("yolo26_obb.jpg", dst, 50)
+}

yolo26_weights/yolo26l-cls.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c1cc0b98541b8404ba796d0402eebeca9e045019d7da08c583e38cf411e5959
+size 56514487

yolo26_weights/yolo26l-obb.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4645afc349167aed3174aac9461b57372b88f98c4485cacf60d8fdf9d9748f73
+size 102910657

yolo26_weights/yolo26l-pose.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:527b2b78daa9a4d969646538294a3445646f647a85ce5ea1049850726457a2b7
+size 104235123

yolo26_weights/yolo26l-seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bba71c343eafbee77a8627fc90ef2bc4b0c01cd8b19bfea376c87ecef35cf361
+size 112265898

yolo26_weights/yolo26l.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca103edb0f50944e5a2467014a2d87a338a5d1a74d0b1d64336938921e0eb5b6
+size 99613333

yolo26_weights/yolo26m-cls.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba396b1496d39e06df773f3afc87f162535a0490c48cf29e19268a6e303ce51e
+size 46567517

yolo26_weights/yolo26m-obb.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6395b66f8b251552baeb959588b72e01f4520c11f2bd953c929be5a47b92ed46
+size 85260285

yolo26_weights/yolo26m-pose.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a96bc1408fdd986522410df331ef258a99c1c68b36bc51646afd1e9031181783
+size 86584943

yolo26_weights/yolo26m-seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0caf9831c09fae8c0848b09f154c689b0d5ab44563ac7c906141aac576ad0817
+size 94615718

yolo26_weights/yolo26m.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ea17cc5173099a3c4b242cc73202af1f656312ad8220b0e442fae97ec06f306
+size 81963155

yolo26_weights/yolo26n-cls.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d635527cf451b4998079ada3d9871798021b71ab1e238fc39a2766216ccdbae0
+size 11293107

yolo26_weights/yolo26n-obb.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0d314ac681df026df63c6922798fd547e98c558d2800355c0fff7a97d327b4e
+size 10207248

yolo26_weights/yolo26n-pose.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e9f5cae96d489e18d2e46ec16807cd1e94dab23b1264d53e80af6796153f1dda
+size 12125854

yolo26_weights/yolo26n-seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e327cc1464524d5bca1974ee56932551604862314bd8ffae92924ae6330f5a61
+size 11218580

yolo26_weights/yolo26n.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:53d5c2fe69bd0524ee02c88e513ceffb0a15f8f040cfc41c9a3c6efc230139ea
+size 9941955

yolo26_weights/yolo26s-cls.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21c0fe906046b38b2ead8c0add802954037580b27b2e3f527a3fe5ea49159614
+size 26925649

yolo26_weights/yolo26s-obb.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9fc512248f3188b793c0d21f00cd74009f9defcfb3a07854994bba7715d6e9f3
+size 39438174

yolo26_weights/yolo26s-pose.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cdc4b440333d482e525148942e456c136a71c127526b2cc80ef6efdbec3505b3
+size 41859220

yolo26_weights/yolo26s-seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e38992c0cc549e1a4ebc8d31e0dbe6da1785eb46c91922aff678ec672ef4da9
+size 41912261

yolo26_weights/yolo26s.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a496b3fc443212340d7966ba8a8c2384447904095a232c6fe2740dda1f22c34
+size 38291128

yolo26_weights/yolo26x-cls.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a31d44ac88dbc78d5efbfa38c2ef135876c3a0276f00e372f5d6ef45682ccf98
+size 118570441

yolo26_weights/yolo26x-obb.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42db72c33ebf04e8ef94ec0bcd576a83895fd69bc117942de634490e76efa0b8
+size 230725483

yolo26_weights/yolo26x-pose.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:afe0819b859a0a12b966d6ce5e673ed9fdc7ad30a39f5c16cde2a192606f9d95
+size 230728671

yolo26_weights/yolo26x-seg.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a76f5817534d996b4009c714021bceffd179d870918daadf6864168b5eb646ee
+size 251680720

yolo26_weights/yolo26x.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a6707f637b0b605a28cc568ebb579233af209cf5183825ae2b713e18b71f13c
+size 223287477