Home
last modified time | relevance | path

Searched refs:box_confidences (Results 1 – 2 of 2) sorted by relevance

/OK3568_Linux_fs/external/rknn-toolkit2/examples/onnx/yolov5/
H A Dtest.py72 def filter_boxes(boxes, box_confidences, box_class_probs): argument
86 box_confidences = box_confidences.reshape(-1)
89 _box_pos = np.where(box_confidences >= OBJ_THRESH)
91 box_confidences = box_confidences[_box_pos]
100 scores = (class_max_score* box_confidences)[_class_pos]
/OK3568_Linux_fs/external/rknn-toolkit2/examples/darknet/yolov3_416x416/
H A Dyolov3_utils.py51 def filter_boxes(boxes, box_confidences, box_class_probs): argument
64 box_scores = box_confidences * box_class_probs