apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bug Fix] fixed labels setting of YOLOv5 (#1213)

Browse Source 
修复自己训练的yolov5无法指定label个数的错误
This commit is contained in:
Zheng-Bicheng
2023-02-02 15:28:38 +08:00
committed by GitHub
parent a711f99c69
commit afa3b886f3
5 changed files with 24 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/vision/detection/contrib/rknpu2/postprocessor.cc
View File

@@ -141,7 +141,6 @@ int RKYOLOPostprocessor::ProcessFP16(float* input, int* anchor, int grid_h,
} else {
limit_score = box_conf_f32 * class_prob_f32;
}
// printf("limit score: %f", limit_score);
if (limit_score > conf_threshold_) {
float box_x, box_y, box_w, box_h;
if (anchor_per_branch_ == 1) {