steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dataset stretch now optional

Browse Source
This commit is contained in:
Si11ium 2020-05-15 09:43:30 +02:00
parent f458660613
commit 09f9dd9131
4 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
_paramters.py
View File

@ -28,6 +28,7 @@ main_arg_parser.add_argument("--data_sr", type=int, default=16000, help="")
main_arg_parser.add_argument("--data_hop_length", type=int, default=256, help="")
main_arg_parser.add_argument("--data_n_fft", type=int, default=512, help="")
main_arg_parser.add_argument("--data_mixup", type=strtobool, default=False, help="")
main_arg_parser.add_argument("--data_stretch", type=strtobool, default=False, help="")
# Transformation Parameters
main_arg_parser.add_argument("--data_loudness_ratio", type=float, default=0, help="") # 0.4

2
datasets/binar_masks.py
View File

@ -19,7 +19,7 @@ class BinaryMasksDataset(Dataset):
def sample_shape(self):
return self[0][0].shape
def __init__(self, data_root, setting, mel_transforms, transforms=None, mixup=False, stretch_dataset=True):
def __init__(self, data_root, setting, mel_transforms, transforms=None, mixup=False, stretch_dataset=False):
self.stretch = stretch_dataset
assert isinstance(setting, str), f'Setting has to be a string, but was: {type(setting)}.'
assert setting in V.DATA_OPTIONS, f'Setting must match one of: {V.DATA_OPTIONS}.'

17
models/residual_conv_classifier.py
View File

@ -39,15 +39,18 @@ class ResidualConvClassifier(BinaryMaskDatasetMixin,
self.conv_list.append(ConvModule(last_shape, self.conv_filters[0], (k, k), conv_stride=(2, 2), conv_padding=1,
**self.params.module_kwargs))
last_shape = self.conv_list[-1].shape
for filters in self.conv_filters:
conv_module_params.update(conv_filters=filters)
for idx in range(len(self.conv_filters)):
conv_module_params.update(conv_filters=self.conv_filters[idx])
self.conv_list.append(ResidualModule(last_shape, ConvModule, 3, **conv_module_params))
last_shape = self.conv_list[-1].shape
self.conv_list.append(ConvModule(last_shape, filters, (k, k), conv_stride=(2, 2), conv_padding=2,
**self.params.module_kwargs))
for param in self.conv_list[-1].parameters():
param.requires_grad = False
last_shape = self.conv_list[-1].shape
try:
self.conv_list.append(ConvModule(last_shape, self.conv_filters[idx+1], (k, k), conv_stride=(2, 2), conv_padding=2,
**self.params.module_kwargs))
for param in self.conv_list[-1].parameters():
param.requires_grad = False
last_shape = self.conv_list[-1].shape
except IndexError:
pass
self.full_1 = LinearModule(self.conv_list[-1].shape, self.params.lat_dim, **self.params.module_kwargs)
self.full_2 = LinearModule(self.full_1.shape, self.full_1.shape * 2, **self.params.module_kwargs)

2
util/module_mixins.py
View File

@ -142,7 +142,7 @@ class BinaryMaskDatasetMixin:
**dict(
# TRAIN DATASET
train_dataset=BinaryMasksDataset(self.params.root, setting=V.DATA_OPTIONS.train,
mixup=self.params.mixup,
mixup=self.params.mixup, stretch_dataset=self.params.stretch,
mel_transforms=mel_transforms_train, transforms=aug_transforms),
# VALIDATION DATASET
val_train_dataset=BinaryMasksDataset(self.params.root, setting=V.DATA_OPTIONS.train,