Searched refs:objProbs (Results 1 – 3 of 3) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/ |
| H A D | post_process.cc | 144 … std::vector<float> &boxes, std::vector<float> &objProbs, std::vector<int> &classId, in process() argument
192 objProbs.push_back(sigmoid(max_class_prob * box_prob)); in process()
219 std::vector<float> objProbs; in post_process() local
228 … stride0, filterBoxes, objProbs, classId, conf_threshold, qnt_zps[0], qnt_scales[0]); in post_process()
236 … stride1, filterBoxes, objProbs, classId, conf_threshold, qnt_zps[1], qnt_scales[1]); in post_process()
244 … stride2, filterBoxes, objProbs, classId, conf_threshold, qnt_zps[2], qnt_scales[2]); in post_process()
261 quick_sort_indice_inverse(objProbs, 0, validCount - 1, indexArray); in post_process()
286 float obj_conf = objProbs[i]; in post_process()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/ |
| H A D | postprocess.cc | 190 …std::vector<float>& boxes, std::vector<float>& objProbs, std::vector<int>& classId, float threshol… in process() argument
225 …objProbs.push_back(sigmoid(deqnt_affine_to_f32(maxClassProbs, zp, scale))* sigmoid(deqnt_affine_to… in process()
257 std::vector<float> objProbs; in post_process() local
265 …ss(input0, (int*)anchor0, grid_h0, grid_w0, model_in_h, model_in_w, stride0, filterBoxes, objProbs, in post_process()
273 …ss(input1, (int*)anchor1, grid_h1, grid_w1, model_in_h, model_in_w, stride1, filterBoxes, objProbs, in post_process()
281 …ss(input2, (int*)anchor2, grid_h2, grid_w2, model_in_h, model_in_w, stride2, filterBoxes, objProbs, in post_process()
295 quick_sort_indice_inverse(objProbs, 0, validCount - 1, indexArray); in post_process()
317 float obj_conf = objProbs[i]; in post_process()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | postprocess.cc | 208 … std::vector<float> &boxes, std::vector<float> &objProbs, std::vector<int> &classId, in process() argument
254 …objProbs.push_back(sigmoid(deqnt_affine_to_f32(maxClassProbs, zp, scale))* sigmoid(deqnt_affine_to… in process()
285 std::vector<float> objProbs; in post_process() local
294 … stride0, filterBoxes, objProbs, classId, conf_threshold, qnt_zps[0], qnt_scales[0]); in post_process()
302 … stride1, filterBoxes, objProbs, classId, conf_threshold, qnt_zps[1], qnt_scales[1]); in post_process()
310 … stride2, filterBoxes, objProbs, classId, conf_threshold, qnt_zps[2], qnt_scales[2]); in post_process()
325 quick_sort_indice_inverse(objProbs, 0, validCount - 1, indexArray); in post_process()
350 float obj_conf = objProbs[i]; in post_process()
|