Merge with local

.gitignore

@@ -13,4 +13,7 @@ lightning_logs/
 *.benchmarks/
 dist/
 examples/*.ipynb_checkpoints/
+<<<<<<< HEAD
 docs/bin
+=======
+>>>>>>> 53cb06343cb89ff1e764c6813fe1d71d981cae0f

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
 ~~~~~~~~~~~~~~
 

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
+

pyproject.toml

@@ -37,9 +37,12 @@ dependencies = [
     "sympy",
     "numba",
     "click",
+<<<<<<< HEAD
     "nbsphinx",
     "pypandoc",
     "opencv-contrib-python-headless",
+=======
+>>>>>>> 53cb06343cb89ff1e764c6813fe1d71d981cae0f
 ]
 dynamic = ["version"]
 

scripts/generate_wideband_sig53.py

@@ -45,7 +45,11 @@ def generate(root: str, configs: List[conf.WidebandSig53Config]):
     "--all", default=True, help="Generate all versions of wideband_sig53 dataset."
 )
 @click.option(
+<<<<<<< HEAD
     "--qa", default=False, help="Generate only QA versions of wideband_sig53 dataset."
+=======
+    "--qa", default=True, help="Generate only QA versions of wideband_sig53 dataset."
+>>>>>>> 53cb06343cb89ff1e764c6813fe1d71d981cae0f
 )
 @click.option(
     "--impaired",

torchsig/__init__.py

@@ -1 +1,5 @@
+<<<<<<< HEAD
 __version__ = "0.4.2"
+=======
+__version__ = "0.4.1"
+>>>>>>> 53cb06343cb89ff1e764c6813fe1d71d981cae0f