from augmentation import Augmentator
from train import train_dataset, convert2rknn

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