* Fix YOLOv11 incorrectly checks hwc

lxowalle · lxowalle · commit 939bb5676a46 · 2026-01-22T20:13:39.000+08:00
diff --git a/components/nn/include/maix_nn_yolo11.hpp b/components/nn/include/maix_nn_yolo11.hpp
@@ -384,7 +384,7 @@ namespace maix::nn
                     nn::LayerInfo &item = outputs[i];
                     if(item.shape.size() == 4)
                     {
-                        if(item.shape[1] == 4 && item.shape[2] == _anchor_num && item.shape[3] == 1) // hwc
+                        if(item.shape[1] == _anchor_num && item.shape[2] == (int)labels.size() && item.shape[3] == 1) // hwc
                         {
                             _out_chw = false;
                             _out_idxes.dfl = i;

Original file line number	Diff line number	Diff line change
`@@ -384,7 +384,7 @@ namespace maix::nn`
`384`	`384`	`nn::LayerInfo &item = outputs[i];`
`385`	`385`	`if(item.shape.size() == 4)`
`386`	`386`	`{`
`387`		`- if(item.shape[1] == 4 && item.shape[2] == _anchor_num && item.shape[3] == 1) // hwc`
	`387`	`+ if(item.shape[1] == _anchor_num && item.shape[2] == (int)labels.size() && item.shape[3] == 1) // hwc`
`388`	`388`	`{`
`389`	`389`	`_out_chw = false;`
`390`	`390`	`_out_idxes.dfl = i;`