speed aug fixed

This commit is contained in:
Si11ium 2020-05-21 14:42:35 +02:00
parent d58bcbf14b
commit e4abbc045c
3 changed files with 15 additions and 17 deletions

View File

@ -34,8 +34,8 @@ main_arg_parser.add_argument("--data_shift_ratio", type=float, default=0.3, help
main_arg_parser.add_argument("--data_noise_ratio", type=float, default=0, help="") # 0.4 main_arg_parser.add_argument("--data_noise_ratio", type=float, default=0, help="") # 0.4
main_arg_parser.add_argument("--data_mask_ratio", type=float, default=0, help="") # 0.2 main_arg_parser.add_argument("--data_mask_ratio", type=float, default=0, help="") # 0.2
main_arg_parser.add_argument("--data_speed_amount", type=float, default=0, help="") # 0.3 main_arg_parser.add_argument("--data_speed_amount", type=float, default=0, help="") # 0.3
main_arg_parser.add_argument("--data_speed_min", type=float, default=0, help="") # 0.7 main_arg_parser.add_argument("--data_speed_min", type=float, default=0.7, help="") # 0.7
main_arg_parser.add_argument("--data_speed_max", type=float, default=0, help="") # 1.7 main_arg_parser.add_argument("--data_speed_max", type=float, default=1.7, help="") # 1.7
# Model Parameters # Model Parameters
main_arg_parser.add_argument("--model_type", type=str, default="RCC", help="") main_arg_parser.add_argument("--model_type", type=str, default="RCC", help="")

View File

@ -51,7 +51,6 @@ class BinaryMasksDataset(Dataset):
additional_dict = ({f'X{key}': val for key, val in labeldict.items()}) additional_dict = ({f'X{key}': val for key, val in labeldict.items()})
additional_dict.update({f'XX{key}': val for key, val in labeldict.items()}) additional_dict.update({f'XX{key}': val for key, val in labeldict.items()})
additional_dict.update({f'XXX{key}': val for key, val in labeldict.items()}) additional_dict.update({f'XXX{key}': val for key, val in labeldict.items()})
additional_dict.update({f'XXXX{key}': val for key, val in labeldict.items()})
labeldict.update(additional_dict) labeldict.update(additional_dict)
# Delete File if one exists. # Delete File if one exists.

View File

@ -30,21 +30,20 @@ if __name__ == '__main__':
all_conf = dict(data_speed_factor=0.7, data_speed_ratio=0.4, data_mask_ratio=0.2, all_conf = dict(data_speed_factor=0.7, data_speed_ratio=0.4, data_mask_ratio=0.2,
data_noise_ratio=0.4, data_shift_ratio=0.4, data_loudness_ratio=0.4, data_noise_ratio=0.4, data_shift_ratio=0.4, data_loudness_ratio=0.4,
data_stretch=True, train_epochs=101) data_stretch=True, train_epochs=101)
speed_conf = dict(data_speed_factor=0.7, data_speed_ratio=0.4, data_mask_ratio=0.0, speed_conf = raw_conf.copy()
data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.0, speed_conf.update(data_speed_amount=0.4, data_speed_min=0.7, data_speed_max=1.7,
data_stretch=True, train_epochs=101) data_stretch=True, train_epochs=101)
mask_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.2, mask_conf = raw_conf.copy()
data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.0, mask_conf.update(data_mask_ratio=0.2, data_stretch=True, train_epochs=101)
data_stretch=True, train_epochs=101)
noise_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0, noise_conf = raw_conf.copy()
data_noise_ratio=0.4, data_shift_ratio=0.0, data_loudness_ratio=0.0, noise_conf.update(data_noise_ratio=0.4, data_stretch=True, train_epochs=101)
data_stretch=True, train_epochs=101)
shift_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0, shift_conf = raw_conf.copy()
data_noise_ratio=0.0, data_shift_ratio=0.4, data_loudness_ratio=0.0, shift_conf.update(data_shift_ratio=0.4, data_stretch=True, train_epochs=101)
data_stretch=True, train_epochs=101)
loudness_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0, loudness_conf = raw_conf.copy()
data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.4, loudness_conf.update(data_loudness_ratio=0.4, data_stretch=True, train_epochs=101)
data_stretch=True, train_epochs=101)
for dicts in [raw_conf, all_conf, speed_conf, mask_conf, noise_conf, shift_conf, loudness_conf]: for dicts in [raw_conf, all_conf, speed_conf, mask_conf, noise_conf, shift_conf, loudness_conf]: