lucytuan commited on
Commit
ccb0db9
·
1 Parent(s): 50c87b9

➕ [Add] dependency for vertical flip

Browse files
Files changed (1) hide show
  1. utils/dataloader.py +1 -1
utils/dataloader.py CHANGED
@@ -10,7 +10,7 @@ from tqdm.rich import tqdm
10
  import diskcache as dc
11
  from typing import Union
12
  from drawer import draw_bboxes
13
- from dataargument import Compose, RandomHorizontalFlip, RandomVerticalFlip, Mosaic
14
 
15
 
16
  class YoloDataset(Dataset):
 
10
  import diskcache as dc
11
  from typing import Union
12
  from drawer import draw_bboxes
13
+ from dataargument import Compose, RandomHorizontalFlip, Mosaic, RandomVerticalFlip
14
 
15
 
16
  class YoloDataset(Dataset):