data package
Subpackages
- data.collate_fns package
- data.datasets package
- Subpackages
- data.datasets.audio_classification package
- data.datasets.classification package
- Submodules
- data.datasets.classification.base_image_classification_dataset module
- data.datasets.classification.base_imagenet_shift_dataset module
- data.datasets.classification.imagenet module
- data.datasets.classification.imagenet_a module
- data.datasets.classification.imagenet_r module
- data.datasets.classification.imagenet_sketch module
- data.datasets.classification.imagenet_synsets module
- data.datasets.classification.imagenet_v2 module
- data.datasets.classification.places365 module
- Module contents
- data.datasets.detection package
- data.datasets.multi_modal_img_text package
- data.datasets.segmentation package
- data.datasets.utils package
- Submodules
- data.datasets.dataset_base module
- Module contents
- Subpackages
- data.loader package
- data.sampler package
- data.text_tokenizer package
- data.transforms package
- Subpackages
- Submodules
- data.transforms.audio module
- data.transforms.audio_bytes module
- data.transforms.base_transforms module
- data.transforms.common module
- data.transforms.image_bytes module
- data.transforms.image_pil module
FixedSizeCropScaleJitterRandomResizedCropAutoAugmentRandAugmentTrivialAugmentWideRandomHorizontalFlipRandomRotateResizeCenterCropSSDCropingPhotometricDistortBoxPercentCoordsInstanceProcessorRandomResizeRandomShortSizeResizeRandomErasingRandomGaussianBlurRandomCropToTensorRandomOrderRandAugmentTimm
- data.transforms.image_torch module
- data.transforms.utils module
- data.transforms.video module
- Module contents
- data.video_reader package
- Submodules
- data.video_reader.base_av_reader module
VideoDurationDoesNotMatchAudioDurationErrorBaseAVReaderBaseAVReader.add_arguments()BaseAVReader.__init__()BaseAVReader.get_frame_transform()BaseAVReader.check_video()BaseAVReader.read_video()BaseAVReader.num_frames()BaseAVReader.random_sampling()BaseAVReader.uniform_sampling()BaseAVReader.read_video_file_into_clips()BaseAVReader.dummy_audio_video_clips()
- data.video_reader.decord_reader module
- data.video_reader.pyav_reader module
- Module contents
Submodules
data.data_loaders module
- data.data_loaders.create_test_loader(opts: Namespace) CVNetsDataLoader[source]
Helper function to create and return a dataset loader for test dataset from command-line arguments
- data.data_loaders.create_train_val_loader(opts: Namespace) Tuple[CVNetsDataLoader, CVNetsDataLoader | None, Sampler][source]
Helper function to create training and validation data loaders.
- Parameters:
opts – Command-line arguments
- Returns:
A tuple containing training data loader, (optional) validation data loader, and training data sampler.
- data.data_loaders.get_num_data_samples_as_int_or_mapping(dataset: BaseDataset) int | Mapping[str, int][source]
Return the number of samples in the dataset.
The dataset can be a single or composition of multiple datasets (as in multi-task learning). For a single dataset, the number of samples is integer while for multiple datasets, a dictionary is returned with task name and number of samples per task.
- Parameters:
dataset – An instance of data.datasets.BaseDataset class
- Returns:
An integer for single dataset and mapping for composite datasets.