diff --git a/manual_run.py b/manual_run.py index 2339377..a0c34fc 100644 --- a/manual_run.py +++ b/manual_run.py @@ -13,12 +13,12 @@ from security import Security # 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) -# +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) +shutil.copy(path.join(model_dir, 'weights', 'best.pt'), model_path) api_client = ApiClient() onnx_path = export_onnx(model_path) diff --git a/requirements.txt b/requirements.txt index 7c6e513..3f8f349 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,4 +19,4 @@ onnxruntime-gpu netron pycuda tensorrt - onnx \ No newline at end of file +onnx \ No newline at end of file