From 4a6e14a31da1c24f4cb919a562ff918f368b7dfe Mon Sep 17 00:00:00 2001 From: gvanhoy Date: Wed, 19 Jul 2023 14:16:07 -0400 Subject: [PATCH 1/2] Fixing QA generation --- torchsig/datasets/conf.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchsig/datasets/conf.py b/torchsig/datasets/conf.py index 93e418a..9bf43d4 100644 --- a/torchsig/datasets/conf.py +++ b/torchsig/datasets/conf.py @@ -124,7 +124,7 @@ class WidebandSig53CleanValConfig(WidebandSig53CleanTrainConfig): @dataclass class WidebandSig53CleanValQAConfig(WidebandSig53CleanValConfig): - num_samples: int = 25 + num_samples: int = 250 @dataclass @@ -137,7 +137,7 @@ class WidebandSig53ImpairedTrainConfig(WidebandSig53Config): @dataclass class WidebandSig53ImpairedTrainQAConfig(WidebandSig53ImpairedTrainConfig): - num_samples: int = 2_50 + num_samples: int = 250 @dataclass @@ -149,4 +149,4 @@ class WidebandSig53ImpairedValConfig(WidebandSig53ImpairedTrainConfig): @dataclass class WidebandSig53ImpairedValQAConfig(WidebandSig53ImpairedValConfig): - num_samples: int = 25 + num_samples: int = 250 From 133d31705af29ae07ac0e40f1cbdc7dc0c91dc65 Mon Sep 17 00:00:00 2001 From: Luke Boegner Date: Sat, 22 Jul 2023 15:43:47 -0400 Subject: [PATCH 2/2] Explicit module updates in dataset and transform RTD files --- docs/datasets.rst | 17 +++- docs/transforms.rst | 223 ++++++++++++++++++++++++++++++-------------- 2 files changed, 168 insertions(+), 72 deletions(-) diff --git a/docs/datasets.rst b/docs/datasets.rst index 903634f..465c22b 100644 --- a/docs/datasets.rst +++ b/docs/datasets.rst @@ -19,7 +19,7 @@ All the datasets have almost similar API. They all have a common argument: ``transform`` to transform the input data. -.. currentmodule:: torchsig.datasets +.. currentmodule:: torchsig.datasets.sig53 Sig53 @@ -28,24 +28,32 @@ Sig53 .. autoclass:: Sig53 +.. currentmodule:: torchsig.datasets.wideband_sig53 + WidebandSig53 ~~~~~~~~~~~~~~ .. autoclass:: WidebandSig53 +.. currentmodule:: torchsig.datasets.modulations + ModulationsDataset ~~~~~~~~~~~~~~~~~~~~ .. autoclass:: ModulationsDataset +.. currentmodule:: torchsig.datasets.wideband + WidebandModulationsDataset ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ .. autoclass:: WidebandModulationsDataset +.. currentmodule:: torchsig.datasets.synthetic + DigitalModulationDataset ~~~~~~~~~~~~~~~~~~~~~~~~~~ @@ -82,6 +90,8 @@ FMDataset .. autoclass:: FMDataset +.. currentmodule:: torchsig.datasets.wideband + WidebandDataset ~~~~~~~~~~~~~~~~~~ @@ -94,12 +104,17 @@ SyntheticBurstSourceDataset .. autoclass:: SyntheticBurstSourceDataset +.. currentmodule:: torchsig.datasets.file_datasets + + FileBurstSourceDataset ~~~~~~~~~~~~~~~~~~~~~~~~ .. autoclass:: FileBurstSourceDataset +.. currentmodule:: torchsig.datasets.radioml + RadioML2016 ~~~~~~~~~~~~~~ diff --git a/docs/transforms.rst b/docs/transforms.rst index 62d41ef..e9e83be 100644 --- a/docs/transforms.rst +++ b/docs/transforms.rst @@ -11,9 +11,9 @@ This is useful if you have to build a more complex transformation pipeline .. contents:: Transforms :local: -General Transforms ------------------- -.. currentmodule:: torchsig.transforms +Transforms +---------- +.. currentmodule:: torchsig.transforms.transforms Transform ^^^^^^^^^ @@ -23,9 +23,9 @@ Compose ^^^^^^^^^ .. autoclass:: Compose -NoTransform -^^^^^^^^^^^^^ -.. autoclass:: NoTransform +Identity +^^^^^^^^^ +.. autoclass:: Identity Lambda ^^^^^^^^^ @@ -51,48 +51,6 @@ TargetConcatenate ^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: TargetConcatenate -RandAugment -^^^^^^^^^^^^^ -.. autoclass:: RandAugment - - -Deep Learning Techniques ------------------------- -.. currentmodule:: torchsig.transforms.deep_learning_techniques.dlt - -DatasetBasebandMixUp -^^^^^^^^^^^^^^^^^^^^^ -.. autoclass:: DatasetBasebandMixUp - -DatasetBasebandCutMix -^^^^^^^^^^^^^^^^^^^^^^^^^ -.. autoclass:: DatasetBasebandCutMix - -CutOut -^^^^^^^^^ -.. autoclass:: CutOut - -PatchShuffle -^^^^^^^^^^^^^ -.. autoclass:: PatchShuffle - -DatasetWidebandMixUp -^^^^^^^^^^^^^^^^^^^^^ -.. autoclass:: DatasetWidebandMixUp - -DatasetWidebandCutMix -^^^^^^^^^^^^^^^^^^^^^^^^^ -.. autoclass:: DatasetWidebandCutMix - -SpectrogramRandomResizeCrop -^^^^^^^^^^^^^^^^^^^^^^^^^^^^ -.. autoclass:: SpectrogramRandomResizeCrop - - -Expert Feature Transforms -------------------------- -.. currentmodule:: torchsig.transforms.expert_feature.eft - InterleaveComplex ^^^^^^^^^^^^^^^^^ .. autoclass:: InterleaveComplex @@ -137,24 +95,51 @@ ReshapeTransform ^^^^^^^^^^^^^^^^^ .. autoclass:: ReshapeTransform - -Signal Processing Transforms ----------------------------- -.. currentmodule:: torchsig.transforms.signal_processing.sp +RandAugment +^^^^^^^^^^^^^ +.. autoclass:: RandAugment Normalize ^^^^^^^^^ .. autoclass:: Normalize + +Augmentations +------------- +.. currentmodule:: torchsig.transforms.transforms + +DatasetBasebandMixUp +^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DatasetBasebandMixUp + +DatasetBasebandCutMix +^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DatasetBasebandCutMix + +CutOut +^^^^^^^^^ +.. autoclass:: CutOut + +PatchShuffle +^^^^^^^^^^^^^ +.. autoclass:: PatchShuffle + +DatasetWidebandMixUp +^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DatasetWidebandMixUp + +DatasetWidebandCutMix +^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DatasetWidebandCutMix + +SpectrogramRandomResizeCrop +^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: SpectrogramRandomResizeCrop + RandomResample ^^^^^^^^^^^^^^^^^ .. autoclass:: RandomResample - -System Impairment Transforms ------------------------------ -.. currentmodule:: torchsig.transforms.system_impairment.si - RandomTimeShift ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: RandomTimeShift @@ -231,11 +216,6 @@ RandomConvolve ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: RandomConvolve - -Wireless Channel Transforms ----------------------------- -.. currentmodule:: torchsig.transforms.wireless_channel.wce - TargetSNR ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: TargetSNR @@ -260,15 +240,6 @@ RandomPhaseShift ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: RandomPhaseShift - -Spectrogram Transforms ----------------------------- -.. currentmodule:: torchsig.transforms.spectrogram_transforms.spec - -SpectrogramResize -^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ -.. autoclass:: SpectrogramResize - SpectrogramDropSamples ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: SpectrogramDropSamples @@ -288,3 +259,113 @@ SpectrogramMosaicCrop SpectrogramMosaicDownsample ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. autoclass:: SpectrogramMosaicDownsample + + +Target Transforms +----------------- +.. currentmodule:: torchsig.transforms.target_transforms + +DescToClassName +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToClassName + +DescToClassNameSNR +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToClassNameSNR + +DescToClassIndex +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToClassIndex + +DescToClassIndexSNR +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToClassIndexSNR + +DescToMask +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToMask + +DescToMaskSignal +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToMaskSignal + +DescToMaskFamily +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToMaskFamily + +DescToMaskClass +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToMaskClass + +DescToSemanticClass +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToSemanticClass + +DescToBBox +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToBBox + +DescToAnchorBoxes +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToAnchorBoxes + +DescPassThrough +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescPassThrough + +DescToBinary +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToBinary + +DescToCustom +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToCustom + +DescToClassEncoding +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToClassEncoding + +DescToWeightedMixUp +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToWeightedMixUp + +DescToWeightedCutMix +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToWeightedCutMix + +DescToBBoxDict +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToBBoxDict + +DescToBBoxSignalDict +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToBBoxSignalDict + +DescToBBoxFamilyDict +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToBBoxFamilyDict + +DescToInstMaskDict +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToInstMaskDict + +DescToSignalInstMaskDict +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToSignalInstMaskDict + +DescToSignalFamilyInstMaskDict +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToSignalFamilyInstMaskDict + +DescToListTuple +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: DescToListTuple + +ListTupleToDesc +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: ListTupleToDesc + +LabelSmoothing +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ +.. autoclass:: LabelSmoothing +