KevinMidboe/immich
1
0
Fork 0
mirror of https://github.com/KevinMidboe/immich.git synced 2025-10-29 17:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: machine learning only take results with > 90% confidence (#1875)

Browse Source
This commit is contained in:
Zack Pollard
2023-02-26 04:02:35 +00:00
committed by GitHub
parent 6c7679714b
commit 3d468c369c
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
machine-learning/src/main.py
View File

@@ -45,10 +45,7 @@ def run_engine(engine, path):
for index, pred in enumerate(predictions): for index, pred in enumerate(predictions):
tags = pred['label'].split(', ') tags = pred['label'].split(', ')
if (index == 0): if (pred['score'] > 0.9):
result = tags
else:
if (pred['score'] > 0.5):
result = [*result, *tags] result = [*result, *tags]
if (len(result) > 1): if (len(result) > 1):