lucytuan commited on
Commit
50c87b9
·
2 Parent(s): f611162 288a464

Merge branch 'DATASET' of github.com:LucyTuan/yolov9mit into DATASET

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, 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, RandomVerticalFlip, Mosaic
14
 
15
 
16
  class YoloDataset(Dataset):