search

arcelien / pba

Efficient Learning of Augmentation Policy Schedules
https://bair.berkeley.edu/blog/2019/06/07/data_aug/
Apache License 2.0
505 stars 86 forks source link

Bounding box dataset #17

Open MattVil opened 5 years ago

MattVil commented 5 years ago

Hi everyone, did you tried PBA on a dataset with bounding boxes? If not what part of the code should I change to work with this type of data?

arcelien commented 5 years ago

Unfortunately I have not tried this yet for object detection. You would need to change the set of augmentations operations to take bounding boxes as input, something like https://arxiv.org/abs/1906.11172.