Does the model use data augmentation strategies when training?

xingyizhou / UniDet

Object detection on multiple datasets with an automatically learned unified label space.

489 stars 58 forks source link

Does the model use data augmentation strategies when training? #10

Open ahong007007 opened 3 years ago

ahong007007 commented 3 years ago

Thank you for your hard work and spiritual welfare. Does the model use data augmentation strategies when training? How to increase the data augmentation strategy?

Thank you very much.

xingyizhou commented 2 years ago

Yes, we use the default detectron2 data augmentation (random resize) specified here. You can increase the sizes here (usually increase from 480-960 helps). You can also enable random crops. See the detectron2 documentations for detauls.