c-l/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-12-29 15:11:58 +00:00
Code Releases 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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
machine-learning/src/main.py
View file

@ -45,11 +45,8 @@ 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 result = [*result, *tags]
else:
if (pred['score'] > 0.5):
result = [*result, *tags]
if (len(result) > 1): if (len(result) > 1):
result = list(set(result)) result = list(set(result))