search

meituan / YOLOv6

YOLOv6: a single-stage object detection framework dedicated to industrial applications.
GNU General Public License v3.0
5.72k stars 1.04k forks source link

Where is best spot to add albumentations augmentation? #446

Closed sarmientoj24 closed 2 years ago

sarmientoj24 commented 2 years ago

I already instantiated albumentations. Where is the best place in the code it should be added?

shensheng272 commented 2 years ago

It should be .yolov6/dataset.py

sarmientoj24 commented 2 years ago

Yes. But where in that spot should I put something like images, labels = self.albumentations(...)

shensheng272 commented 2 years ago

https://github.com/meituan/YOLOv6/blob/main/yolov6/data/datasets.py#L141 either before/after/replaceimg, labels = random_affine(...)