manual_run polishing

This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 18:42:26 +03:00
parent 538dc8efa9
commit 80c2433141
+3 -8
View File
@@ -1,18 +1,13 @@
import shutil import shutil
from datetime import datetime
from os import path from os import path
import train import train
from constants import models_dir, prefix, date_format, MODELS_FOLDER, CURRENT_ONNX_MODEL from constants import models_dir, prefix
from api_client import ApiClient
from augmentation import Augmentator from augmentation import Augmentator
from exports import export_onnx
from security import Security
# Augmentator().augment_annotations() # Augmentator().augment_annotations()
#train_dataset(from_scratch=True) # train.train_dataset()
# train.resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')
# resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')
result_dir = '/azaion/dev/ai-training/runs/detect/train12' result_dir = '/azaion/dev/ai-training/runs/detect/train12'
model_dir = path.join(models_dir, f'{prefix}2025-05-18') model_dir = path.join(models_dir, f'{prefix}2025-05-18')