Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-07-30 13:48:07 +02:00
Code Issues Projects Releases Wiki Activity

detectors: simplified/optimized yolov8_postprocess

Browse Source
This commit is contained in:
Indrek Mandre 2024-01-28 11:14:37 +02:00
parent 2634348909
commit de02ba761b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/detectors/yolo_utils.py
View File

@ -18,7 +18,7 @@ def yolov8_postprocess(model_input_shape, tensor_output, box_count = 20):
model_class_count = tensor_output.shape[1] - 4 model_class_count = tensor_output.shape[1] - 4
probs = tensor_output[0, 4:, :] probs = tensor_output[0, 4:, :]
all_ids = np.argmax(probs, axis=0) all_ids = np.argmax(probs, axis=0)
all_confidences = np.take(probs.T, model_class_count*np.arange(0, model_box_count) + all_ids) all_confidences = probs.T[np.arange(model_box_count), all_ids]
all_boxes = tensor_output[0, 0:4, :].T all_boxes = tensor_output[0, 0:4, :].T
mask = (all_confidences > 0.30) mask = (all_confidences > 0.30)
class_ids = all_ids[mask] class_ids = all_ids[mask]