Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

detector/yolo_utils: indentation, remove unused variable

Browse Source
This commit is contained in:
Indrek Mandre 2024-01-28 13:01:50 +02:00
parent de02ba761b
commit da31ee197c
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
frigate/detectors/yolo_utils.py
View File

@ -15,7 +15,6 @@ def yolov8_preprocess(tensor_input, model_input_shape):
def yolov8_postprocess(model_input_shape, tensor_output, box_count = 20): def yolov8_postprocess(model_input_shape, tensor_output, box_count = 20):
model_box_count = tensor_output.shape[2] model_box_count = tensor_output.shape[2]
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 = probs.T[np.arange(model_box_count), all_ids] all_confidences = probs.T[np.arange(model_box_count), all_ids]