Doni/safesight
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update YOLO node to use cxcywh box format for V8 box processing

Browse Source
This commit is contained in:
sladro 2026-02-27 19:49:56 +08:00
parent bbf987fe87
commit b4709b20cd
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/test_cam1_ppe11_yolo_debug.json
View File

@ -49,7 +49,7 @@
"model_w": 768,
"model_h": 768,
"num_classes": 11,
"v8_box_format": "auto",
"v8_box_format": "cxcywh",
"v8_cls_activation": "auto",
"conf": 0.25,
"nms": 0.45,

4
plugins/ai_yolo/ai_yolo_node.cpp
View File

@ -561,7 +561,7 @@ public:
model_input_h_ = config.ValueOr<int>("model_h", 640);
num_classes_ = config.ValueOr<int>("num_classes", 80);
{
const std::string bf = config.ValueOr<std::string>("v8_box_format", "auto");
const std::string bf = config.ValueOr<std::string>("v8_box_format", "cxcywh");
if (bf == "xyxy") {
v8_box_format_ = V8BoxFormat::XyXy;
} else if (bf == "xywh") {
@ -1101,7 +1101,7 @@ private:
int model_input_w_ = 640;
int model_input_h_ = 640;
int num_classes_ = 80;
V8BoxFormat v8_box_format_ = V8BoxFormat::Auto;
V8BoxFormat v8_box_format_ = V8BoxFormat::CxCyWh;
V8ClsActivation v8_cls_activation_ = V8ClsActivation::Auto;
YoloVersion yolo_version_ = YoloVersion::V8;
bool auto_detect_version_ = false;