administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-04 02:05:21 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Change minimum threshold to improve model performance

Browse Source
This commit is contained in:
Anil Ozyalcin 2023-02-16 22:43:50 -08:00
parent e7436a413c
commit 147a90f84a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/plugins/openvino.py
View File

@ -144,7 +144,7 @@ class OvDetector(DetectionApi):
# add scores to the last column # add scores to the last column
dets = np.concatenate((output_data, scores), axis=1) dets = np.concatenate((output_data, scores), axis=1)
# filter out lines with scores below threshold # filter out lines with scores below threshold
dets = dets[dets[:, -1] > 0.3, :] dets = dets[dets[:, -1] > 0.5, :]
# limit to top 20 scores, descending order # limit to top 20 scores, descending order
ordered = dets[dets[:, -1].argsort()[::-1]][:20] ordered = dets[dets[:, -1].argsort()[::-1]][:20]
detections = np.zeros((20, 6), np.float32) detections = np.zeros((20, 6), np.float32)
@ -172,7 +172,7 @@ class OvDetector(DetectionApi):
out_tensor = infer_request.get_output_tensor() out_tensor = infer_request.get_output_tensor()
output_data = out_tensor.data[0] output_data = out_tensor.data[0]
# filter out lines with scores below threshold # filter out lines with scores below threshold
conf_mask = (output_data[:, 4] >= 0.3).squeeze() conf_mask = (output_data[:, 4] >= 0.5).squeeze()
output_data = output_data[conf_mask] output_data = output_data[conf_mask]
# limit to top 20 scores, descending order # limit to top 20 scores, descending order
ordered = output_data[output_data[:, 4].argsort()[::-1]][:20] ordered = output_data[output_data[:, 4].argsort()[::-1]][:20]