Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Torchaug
Torchaug

Contents:

  • Tutorials
    • Transforms Tutorial
    • TA Tensors Tutorial
    • Nested Tensors Tutorial
    • Coco Dataset example
      • COCO dataset example using batch tensors
      • COCO dataset example using nested tensors
  • Transforms
    • Augment
      • CutMix
      • MixUp
      • RandomErasing
      • JPEG
    • Auto-Augment
      • AugMix
      • AutoAugment
      • RandAugment
      • TrivialAugmentWide
    • Color
      • ColorJitter
      • Grayscale
      • RandomAdjustSharpness
      • RandomAutocontrast
      • RandomChannelPermutation
      • RandomColorJitter
      • RandomEqualize
      • RandomGrayscale
      • RandomInvert
      • RandomPhotometricDistort
      • RandomPosterize
      • RandomSolarize
      • RGB
    • Container
      • Compose
      • RandomApply
      • RandomChoice
      • RandomOrder
      • SequentialTransform
    • Geometry
      • CenterCrop
      • ElasticTransform
      • FiveCrop
      • Pad
      • RandomAffine
      • RandomCrop
      • RandomHorizontalFlip
      • RandomIoUCrop
      • RandomPerspective
      • RandomResize
      • RandomResizedCrop
      • RandomRotation
      • RandomShortestSize
      • RandomVerticalFlip
      • RandomZoomOut
      • Resize
      • ScaleJitter
      • TenCrop
    • Meta
      • ClampBoundingBoxes
      • ConvertBoundingBoxFormat
    • Misc
      • GaussianBlur
      • Identity
      • Lambda
      • LinearTransformation
      • Normalize
      • RandomGaussianBlur
      • SanitizeBoundingBoxes
      • ToDtype
    • Temporal
      • UniformTemporalSubsample
    • Transform
      • RandomApplyTransform
      • Transform
    • Type Conversion
      • ToBatchImages
      • ToImage
      • ToPureTensor
  • Functional
    • Augment
      • erase
      • erase_image
      • erase_video
      • jpeg
      • jpeg_image
      • jpeg_video
    • Color
      • adjust_brightness
      • adjust_brightness_batch
      • adjust_contrast
      • adjust_contrast_batch
      • adjust_hue
      • adjust_hue_batch
      • adjust_saturation
      • adjust_saturation_batch
      • adjust_sharpness
      • autocontrast
      • equalize
      • grayscale_to_rgb
      • invert
      • permute_channels
      • posterize
      • rgb_to_grayscale
      • solarize
    • Geometry
      • affine
      • center_crop
      • crop
      • elastic
      • elastic_batch
      • five_crop
      • horizontal_flip
      • pad
      • perspective
      • resize
      • resized_crop
      • rotate
      • ten_crop
      • vertical_flip
    • Meta
      • clamp_bounding_boxes
      • convert_bounding_box_format
      • get_dimensions
      • get_num_channels
      • get_num_frames
      • get_size
    • Misc
      • gaussian_blur
      • gaussian_blur_batch
      • normalize
      • to_dtype
    • Temporal
      • uniform_temporal_subsample
    • Type Conversion
      • to_batch_images
      • to_image
  • TA Tensors
    • TA Tensor
      • TATensor
    • Image
      • Image
      • BatchImages
    • Video
      • Video
      • BatchVideos
    • Mask
      • Mask
      • BatchMasks
      • convert_batch_masks_to_masks
      • convert_masks_to_batch_masks
    • Bounding Boxes
      • BoundingBoxFormat
      • BoundingBoxes
      • BatchBoundingBoxes
      • convert_batch_bboxes_to_bboxes
      • convert_bboxes_to_batch_bboxes
    • Labels
      • Labels
      • BatchLabels
      • convert_batch_labels_to_labels
      • convert_labels_to_batch_labels
    • TA Nested Tensors
      • TANestedTensors
      • ImageNestedTensors
      • LabelsNestedTensors
      • BoundingBoxesNestedTensors
      • MaskNestedTensors
      • NestedTensors
      • VideoNestedTensors
  • Data
    • Datasets
      • wrap_dataset_for_transforms_v2
    • DataLoader
      • default_collate
  • Torchaug speed
Back to top
View this page
Edit this page

DatasetsΒΆ

wrap_dataset_for_transforms_v2(dataset[, ...])

Wrap a torchvision.dataset for usage with torchaug.transforms.
Next
wrap_dataset_for_transforms_v2
Previous
Data
Copyright © 2023-present, Torchaug authors
Made with Sphinx and @pradyunsg's Furo