Searched refs:box_class_probs (Results 1 – 2 of 2) sorted by relevance
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/onnx/yolov5/ |
| H A D | test.py | 52 box_class_probs = sigmoid(input[..., 5:])
69 return box, box_confidence, box_class_probs
72 def filter_boxes(boxes, box_confidences, box_class_probs): argument
87 box_class_probs = box_class_probs.reshape(-1, box_class_probs.shape[-1])
92 box_class_probs = box_class_probs[_box_pos]
94 class_max_score = np.max(box_class_probs, axis=-1)
95 classes = np.argmax(box_class_probs, axis=-1)
|
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/darknet/yolov3_416x416/ |
| H A D | yolov3_utils.py | 30 box_class_probs = sigmoid(input[..., 5:])
49 return box, box_confidence, box_class_probs
51 def filter_boxes(boxes, box_confidences, box_class_probs): argument
64 box_scores = box_confidences * box_class_probs
|