IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
https://detrex.readthedocs.io/en/latest/
Apache License 2.0
2.01k stars 209 forks source link

Using Albumentations for more advanced transforms #206

Closed fernandorovai closed 1 year ago

fernandorovai commented 1 year ago

Had anyone added albumentations to apply rotation and other different transforms on the dataloader? Any idea on what should be the easiest way to implement that? Since detrex uses lazy configs, I don't know if has an easy way to add this feature.

rentainhe commented 1 year ago

Had anyone added albumentations to apply rotation and other different transforms on the dataloader? Any idea on what should be the easiest way to implement that? Since detrex uses lazy configs, I don't know if has an easy way to add this feature.

Maybe you can refer to d2's documentation for more details about the augmentation used in detectron2~ and lazyconfig is just a single wrap of the augmentation module, all you need to do is implement your specific augmentation and wrap it with LazyCall in LazyConfig~ @fernandorovai