from augmentation import Augmentator
from train import train_dataset, convert2rknn

Augmentator().augment_annotations(from_scratch=True)
train_dataset(from_scratch=True)