azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 14:46:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

train fixes merge

Browse Source
This commit is contained in:
zxsanny
2025-05-22 16:56:09 +03:00
parent ce29d7fc58
commit 5b90320e27
3 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
preprocess-train.py
+6 -3
View File
@@ -1,5 +1,8 @@
from augmentation import Augmentator
from train import train_dataset, convert2rknn
from train import train_dataset, convert2rknn, resume_training
Augmentator().augment_annotations()
train_dataset(from_scratch=True)
# Augmentator().augment_annotations()
#train_dataset(from_scratch=True)
resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')