tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)
https://arxiv.org/abs/1904.01355
Other
3.26k stars 630 forks source link

how to add image augmentations #316

Open gitTaner opened 3 years ago

gitTaner commented 3 years ago

transform = T.Compose( [ T.Resize(min_size, max_size), T.RandomHorizontalFlip(flip_prob), T.ToTensor(), normalize_transform, ] there are only aumentations like image resize, rotate etc i want to add some augmentations about image color, how to achieve it? thanks

pp00704831 commented 3 years ago

Hello, I find it in the maskrcnn transforms : https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/maskrcnn_benchmark/data/transforms/transforms.py