rk1808/yolov5-inferrence
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rknn filter class id = 4

Browse Source
This commit is contained in:
asxalex
2025-08-29 16:37:20 +08:00
parent 4f26f54017
commit 4ef4defc6d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rknn.cpp
+1 -1
View File
@@ -119,7 +119,7 @@ int print_object_detect_result(rknn_app_context_t *rknn, object_detect_result_li
continue; continue;
} }
char *label = labels[group->results[i].cls_id]; char *label = labels[group->results[i].cls_id];
if (group->results[i].prop < conf_threshold) { if (group->results[i].cls_id != 5 && group->results[i].prop < conf_threshold) {
printf("skip prop = %f for class %s\n", group->results[i].prop, label); printf("skip prop = %f for class %s\n", group->results[i].prop, label);
continue; continue;
} }