:sparkles: support yolov11

.gitattributes

@@ -38,6 +38,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.so filter=lfs diff=lfs merge=lfs -text
 *.dylib filter=lfs diff=lfs merge=lfs -text
 *.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text
 *.wav filter=lfs diff=lfs merge=lfs -text
 *.txt filter=lfs diff=lfs merge=lfs -text
 *.mvn filter=lfs diff=lfs merge=lfs -text

README.md

@@ -82,3 +82,195 @@ func main() {
 | 原图                                                  | Mask图                                                      |
 |-----------------------------------------------------|------------------------------------------------------------|
 | <img width="100%" src="./examples/test.png" alt=""> | <img width="100%" src="./examples/output_mask.png" alt=""> |
+
+### yolov11-det
+
+```go
+package main
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolov11"
+	"github.com/up-zero/gotool/imageutil"
+	"image"
+	"image/color"
+	"image/draw"
+	"log"
+)
+
+func main() {
+	engine, err := yolov11.NewDetEngine(yolov11.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("yolov11_det.jpg", targetImg, 50)
+}
+```
+
+| 原图                                                  | 检测结果                                                       |
+|-----------------------------------------------------|------------------------------------------------------------|
+| <img width="100%" src="./examples/test.png" alt=""> | <img width="100%" src="./examples/yolov11_det.jpg" alt=""> |
+
+
+### yolov11-seg
+
+```go
+package main
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolov11"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+)
+
+func main() {
+	engine, err := yolov11.NewSegEngine(yolov11.DefaultSegConfig())
+	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("yolov11_seg_mask_%d.png", idx), res.Mask, 100)
+	}
+}
+```
+
+| 原图                                                  | Mask图                                                             |
+|-----------------------------------------------------|-------------------------------------------------------------------|
+| <img width="100%" src="./examples/test.png" alt=""> | <img width="100%" src="./examples/yolov11_seg_mask_0.png" alt=""> |
+
+### yolov11-cls
+
+```go
+package main
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolov11"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+)
+
+func main() {
+	engine, err := yolov11.NewClsEngine(yolov11.DefaultClsConfig())
+	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.70231
+//  Class: 436, Score: 0.22404
+//  Class: 511, Score: 0.02263
+//  Class: 675, Score: 0.01796
+//  Class: 717, Score: 0.00385
+```
+
+### yolov11-pose
+
+```go
+package main
+
+import (
+	"github.com/getcharzp/go-vision/yolov11"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+)
+
+func main() {
+	engine, err := yolov11.NewPoseEngine(yolov11.DefaultPoseConfig())
+	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 := yolov11.DrawPoseResult(img, results)
+	imageutil.Save("yolov11_pose.jpg", dst, 50)
+}
+```
+
+| 原图                                                    | 姿态图                                                         |
+|-------------------------------------------------------|-------------------------------------------------------------|
+| <img width="100%" src="./examples/person.jpg" alt=""> | <img width="100%" src="./examples/yolov11_pose.jpg" alt=""> |
+
+### yolov11-obb
+
+```go
+package main
+
+import (
+	"github.com/getcharzp/go-vision/yolov11"
+	"github.com/up-zero/gotool/imageutil"
+	"image"
+	"image/color"
+	"image/draw"
+	"log"
+)
+
+func main() {
+	engine, err := yolov11.NewOBBEngine(yolov11.DefaultOBBConfig())
+	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("yolov11_obb.jpg", dst, 50)
+}
+```
+
+| 原图                                                  | OBB图                                                       |
+|-----------------------------------------------------|------------------------------------------------------------|
+| <img width="100%" src="./examples/ship.jpg" alt=""> | <img width="100%" src="./examples/yolov11_obb.jpg" alt=""> |

examples/yolov11_test.go

@@ -0,0 +1,130 @@
+package examples
+
+import (
+	"fmt"
+	"github.com/getcharzp/go-vision/yolov11"
+	"github.com/up-zero/gotool/imageutil"
+	"image"
+	"image/color"
+	"image/draw"
+	"testing"
+)
+
+func TestYOLOv11Det(t *testing.T) {
+	cfg := yolov11.DefaultDetConfig()
+	cfg.ModelPath = "../yolov11_weights/yolo11m.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolov11.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("yolov11_det.jpg", targetImg, 50)
+}
+
+func TestYOLOv11Seg(t *testing.T) {
+	cfg := yolov11.DefaultSegConfig()
+	cfg.ModelPath = "../yolov11_weights/yolo11m-seg.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolov11.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("yolov11_seg_mask_%d.png", idx), res.Mask, 100)
+	}
+}
+
+func TestYOLOv11Cls(t *testing.T) {
+	cfg := yolov11.DefaultClsConfig()
+	cfg.ModelPath = "../yolov11_weights/yolo11m-cls.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolov11.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 TestYOLOv11Pose(t *testing.T) {
+	cfg := yolov11.DefaultPoseConfig()
+	cfg.ModelPath = "../yolov11_weights/yolo11m-pose.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolov11.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 := yolov11.DrawPoseResult(img, results)
+	imageutil.Save("yolov11_pose.jpg", dst, 50)
+}
+
+func TestYOLOv11OBB(t *testing.T) {
+	cfg := yolov11.DefaultOBBConfig()
+	cfg.ModelPath = "../yolov11_weights/yolo11m-obb.onnx"
+	cfg.OnnxRuntimeLibPath = "../lib/onnxruntime.dll"
+
+	engine, err := yolov11.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("yolov11_obb.jpg", dst, 50)
+}

yolov11_weights/yolo11l-cls.onnx

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

yolov11_weights/yolo11l-obb.onnx

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

yolov11_weights/yolo11l-pose.onnx

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

yolov11_weights/yolo11l-seg.onnx

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

yolov11_weights/yolo11l.onnx

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

yolov11_weights/yolo11m-cls.onnx

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

yolov11_weights/yolo11m-obb.onnx

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

yolov11_weights/yolo11m-pose.onnx

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

yolov11_weights/yolo11m-seg.onnx

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

yolov11_weights/yolo11m.onnx

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

yolov11_weights/yolo11n-cls.onnx

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

yolov11_weights/yolo11n-obb.onnx

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

yolov11_weights/yolo11n-pose.onnx

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

yolov11_weights/yolo11n-seg.onnx

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

yolov11_weights/yolo11n.onnx

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

yolov11_weights/yolo11s-cls.onnx

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

yolov11_weights/yolo11s-obb.onnx

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

yolov11_weights/yolo11s-pose.onnx

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

yolov11_weights/yolo11s-seg.onnx

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

yolov11_weights/yolo11s.onnx

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

yolov11_weights/yolo11x-cls.onnx

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

yolov11_weights/yolo11x-obb.onnx

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

yolov11_weights/yolo11x-pose.onnx

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

yolov11_weights/yolo11x-seg.onnx

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

yolov11_weights/yolo11x.onnx

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