Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-11-03 19:17:11 -05: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
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):
tags = pred['label'].split(', ')
if (index == 0):
result = tags
else:
if (pred['score'] > 0.5):
result = [*result, *tags]
if (pred['score'] > 0.9):
result = [*result, *tags]
if (len(result) > 1):
result = list(set(result))