diff --git a/main.py b/main.py index cf45999..66613f8 100644 --- a/main.py +++ b/main.py @@ -166,7 +166,7 @@ def _executar_treino_sync(ambiente: str, pular_triagem: bool, forcar_do_zero: bo imagens = [f for f in os.listdir(img_dir) if not f.endswith('.txt')] if len(imagens) > 0: - import random, shutil + import random random.shuffle(imagens) corte = max(1, int(len(imagens) * 0.8)) val_imgs = imagens[corte:]