From 8bb5b515741473163c6f56344702ed7e4eda0917 Mon Sep 17 00:00:00 2001 From: sladro Date: Thu, 8 Jan 2026 10:00:10 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BA=BA=E8=84=B8=E6=A3=80?= =?UTF-8?q?=E8=BD=A6=EF=BC=8C=E6=9C=80=E6=96=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- configs/test_cam1_face_det_rtsp_server.json | 4 ++-- models/RetinaFace_mobile320.rknn | Bin 1641861 -> 1641797 bytes plugins/ai_face_det/ai_face_det_node.cpp | 10 ++++++---- 3 files changed, 8 insertions(+), 6 deletions(-) diff --git a/configs/test_cam1_face_det_rtsp_server.json b/configs/test_cam1_face_det_rtsp_server.json index 2c9716f..7d51bf2 100644 --- a/configs/test_cam1_face_det_rtsp_server.json +++ b/configs/test_cam1_face_det_rtsp_server.json @@ -24,8 +24,8 @@ "type": "preprocess", "role": "filter", "enable": true, - "dst_w": 640, - "dst_h": 640, + "dst_w": 320, + "dst_h": 320, "dst_format": "rgb", "keep_ratio": false, "use_rga": true diff --git a/models/RetinaFace_mobile320.rknn b/models/RetinaFace_mobile320.rknn index 66445a6b1fb171651ecab3d5c41a85d6ed254d09..1e60a95f7f76828ccada407d895b910cf7b18303 100644 GIT binary patch delta 2237 zcmaJ>-Ahwp7(ZvaX@GwbH;VrXTxKIo#-E!3+}EU*tW+DMDaYEXA2XCPvf7B)(# zE)K*l1ceiQEYrdWF@?ZCAo3Wkx+of?Xy~%{ywCGE&)M*>-=6Kf=l6Sl&+|U->3q~O z_OPYk=;8W$fn7HJi(R9((!M}c+@9@YfAq!PV2w4zR=azL4p=J`sY-IRCe`ROWA&@X;+1$e>&uhsGK2=+wwwQK}aIA*HsYKvNF5c>>`7OA%j8x zZ)LDCWHDqjtYff8+w6|n2eNKPza;)G2$eT(F6TE3g0|;JRSAo9x$uW`^ER)3UkNYr z^$^~dFb-PJ{nPnVvlaX~ye$wvG!48_ACIyyicd%w6SwYi1D=Jim8%WRSl|KY^Fl5h z{{)eE6f1w9bsPEhh29yCk%E`(}puHM8&@lh_;A#U@0d*kWj6x_({KRSC ztDkEO28iE~Tf__StBfz}tu_h#+6^s??|5$zm=WPYXE`MH`Ouk>bZmoB^*{51|j6PWOux)2$HE;!qaZ6rROWzR9}f{7nPzA#YBC!Q^{^TsZ- zKoA8qzM33JHwtK5R|NPIX1D$U+FA`4q*Ni13?tfOa|N}&|z8AfQozx-;z{7EApkBBisPxPc?7?XTT&7 zi$VojEwvC@6Tl9Iih-vhpAu7?8-kNCb^Wqlz)JoV12lw6BfLyBlL0pxm~!Zoj9FPo z*@BZ`L^7SE$V=RU`?NF7H>&l zjS7NTfo2kLqk!+;l{Gt{a=w1ntGY(|9SrLkau{+M@)(>98yNB#3K$9*iWrI+N}?nE I8z;{E2kAYqZ2$lO delta 2281 zcmaJ?OH30{6n#SrMJlBVen?ylE-VXZ3&w!>X@e%H7}Bs|L6jjv;*KWNpousohF~Hg zRD^=KD2arykWV7iAPc%cw#Wu?p%YzbG_fRV1V7O?bMGz8FySSqx0#u9?z#8AH?#Pt zsdKm~x3;>bhVhl5{;12AAB}Vi#hKUnfAwZm7`1yNfz#=O{E;V-$fw zayz;TvjmPb4hx5sBb_6I!^V-xv4&8$(kj3| z;`6WCb-w~pqU#u=Cv!CZdRMh&JHHX{2U>v_yq8pUEmKF&hh*)fdoV z0i!`T6wn*xe3`eH5>qV07W%JN>MAGzKBEpOZ6GA!b>dg8;Rvnz0t7S=o&)1bM$#&) zGa@2In-c`Q34II6faDm^I^%2rA&J2ve$^W80jquwjo@?Z0WiMs>87j+5s_kabWqzw z%G{)`1C1!4RNvBgQ*~f~cobdvIKndh2&jOmIxrBJkWoMYWfJh_r+`*j8q_}u#Y zFVKMza|1RH7?^6%kB|zWP1xH!3{EI9#jzDI;QwG=0d2zIW8lXaEaEREafI^%HyTaM z*nQD6qjr}z@01J2Yppu}W%DY0Z^48VGB-gdATR`cqlx4!nrQ#5b&Ff%=Z&ud?U%7w zCnny2sNi)G<0K07mVYM!?T(QzAzu`LNciQ zGw_pf=zx^B0{aN_?=-N99WV&Apn&MspXEmEkl%sx|BIc(>(l}H2#k=?`KkPjDp21-+K1wYc?BLXwlMR? zt_GAl$j7%F70`kP*swant^BDD&?byw1(HF)g8~a7-|hye0|tI_1_}_LMlC&!CZAL? zApV956obGd3iyjxbuXg=nAFxO`lIsIvwk+>w|0fc?HqeKvN;?aIUG)oeH{Baayjxi Q4saagI1~?$A6~rnA4G@4NdN!< diff --git a/plugins/ai_face_det/ai_face_det_node.cpp b/plugins/ai_face_det/ai_face_det_node.cpp index 144600c..fd1ebb2 100644 --- a/plugins/ai_face_det/ai_face_det_node.cpp +++ b/plugins/ai_face_det/ai_face_det_node.cpp @@ -165,12 +165,14 @@ bool ExtractNc(const Tensor& t, int c, NcTensor& out) { // Common: [1, C, N] or [1, N, C] const uint32_t d1 = t.dims[1]; const uint32_t d2 = t.dims[2]; - if (static_cast(d1) == c) { - n = static_cast(d2); - transposed = true; // CxN - } else if (static_cast(d2) == c) { + if (static_cast(d2) == c) { n = static_cast(d1); transposed = false; // NxC + } else if (static_cast(d1) == c) { + n = static_cast(d2); + transposed = true; // CxN + } else { + return false; // 明确拒绝,不 fallback } } else if (t.dims.size() == 2) { // [N, C] or [C, N]