Skip to content

Instantly share code, notes, and snippets.

@farukcankaya

farukcankaya/multimodal_aug_aug_input.py

Created October 20, 2022 07:41
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/17958cd50d07e55aae71305e2119b5cf to your computer and use it in GitHub Desktop.
Save farukcankaya/17958cd50d07e55aae71305e2119b5cf to your computer and use it in GitHub Desktop.
Download ZIP
Raw
multimodal_aug_aug_input.py
class AugInput:
def transform(self, tfm: Transform) -> None:
self.image = tfm.apply_image(self.image)
if self.boxes is not None:
self.boxes = tfm.apply_box(self.boxes)
if self.sem_seg is not None:
self.sem_seg = tfm.apply_segmentation(self.sem_seg)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment