mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-07-26 13:47:03 +02:00
detectors: simplified/optimized yolov8_postprocess
This commit is contained in:
parent
2634348909
commit
de02ba761b
@ -18,7 +18,7 @@ def yolov8_postprocess(model_input_shape, tensor_output, box_count = 20):
|
||||
model_class_count = tensor_output.shape[1] - 4
|
||||
probs = tensor_output[0, 4:, :]
|
||||
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
|
||||
mask = (all_confidences > 0.30)
|
||||
class_ids = all_ids[mask]
|
||||
|
Loading…
Reference in New Issue
Block a user