refactor augmentation to class, update classes.json, fix small bugs

This commit is contained in:
zxsanny
2025-05-15 06:10:10 +03:00
parent fd4cd75265
commit 292809593c
5 changed files with 188 additions and 190 deletions
+3 -4
View File
@@ -1,6 +1,5 @@
from preprocessing import preprocess_annotations
from augmentation import Augmentator
from train import train_dataset, convert2rknn
preprocess_annotations()
train_dataset('2024-10-01')
convert2rknn()
Augmentator().augment_annotations(from_scratch=True)
train_dataset(from_scratch=True)