mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 19:26:36 +00:00
Alex Bezdieniezhnykh
25 lines
802 B
Python
25 lines
802 B
Python
import shutil
|
|
from datetime import datetime
|
|
from os import path
|
|
|
|
import train
|
|
from constants import models_dir, prefix, date_format, MODELS_FOLDER, CURRENT_ONNX_MODEL
|
|
from api_client import ApiClient
|
|
from augmentation import Augmentator
|
|
from exports import export_onnx
|
|
from security import Security
|
|
|
|
# Augmentator().augment_annotations()
|
|
#train_dataset(from_scratch=True)
|
|
|
|
# resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')
|
|
|
|
result_dir = '/azaion/dev/ai-training/runs/detect/train12'
|
|
model_dir = path.join(models_dir, f'{prefix}2025-05-18')
|
|
shutil.copytree(result_dir, model_dir, dirs_exist_ok=True)
|
|
|
|
model_path = path.join(models_dir, f'{prefix[:-1]}.pt')
|
|
shutil.copy(path.join(model_dir, 'weights', 'best.pt'), model_path)
|
|
|
|
train.export_current_model()
|
|
print('success!') |