Skip to content

Instantly share code, notes, and snippets.

@farukcankaya

farukcankaya/multimodal_aug_dataset_mapper.py

Created October 20, 2022 00:48
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save farukcankaya/284c125419c1d308051e9a17d7f4df3d to your computer and use it in GitHub Desktop.
Save farukcankaya/284c125419c1d308051e9a17d7f4df3d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
multimodal_aug_dataset_mapper.py
# from: https://github.com/facebookresearch/detectron2/blob/998c4e15220a0db9f172b1e7ecf376a59b873f1d/detectron2/data/dataset_mapper.py#L163-L165
aug_input = T.AugInput(image, sem_seg=sem_seg_gt)
transforms = self.augmentations(aug_input)
image, sem_seg_gt = aug_input.image, aug_input.sem_seg
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment