fix: remove aprovação automática de imagens sem detecção na triagem

Imagens onde o YOLO não detecta nada (conf == 0) passavam direto para
base-oficial, contaminando o dataset. Agora vão para descartados e
precisam de revisão humana antes de entrar no treino.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Davi Dias
2026-06-25 19:51:04 -03:00
co-authored by Claude Sonnet 4.6
parent c8ce09e5af
commit 432159bb2e
2 changed files with 30 additions and 1 deletions
+1 -1
View File
@@ -115,7 +115,7 @@ def _executar_treino_sync(ambiente: str, pular_triagem: bool) -> dict:
pred = modelo_base(img_data, conf=0.1, verbose=False)
confs = [float(b.conf) for r in pred for b in r.boxes]
media = sum(confs)/len(confs) if confs else 0
decisao = "base-oficial" if media >= 0.15 or media == 0 else "descartados"
decisao = "base-oficial" if media >= 0.15 else "descartados"
log_print(f"{os.path.basename(img_key)}: Conf. {media:.2f} -> {decisao}")
for k in [img_key, txt_key]:
try: