MCPcopy Create free account

hub / github.com/RetroCirce/HTS-Audio-Transformer / functions

Functions248 in github.com/RetroCirce/HTS-Audio-Transformer

↓ 99 callersFunctioninit_layer
Initialize a Linear or Convolutional layer.
models.py:11
↓ 65 callersFunctioninit_bn
Initialize a Batchnorm layer.
models.py:20
↓ 41 callersMethod__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:401
↓ 32 callersFunctiondo_mixup
Args: x: (batch_size , ...) mixup_lambda: (batch_size,) Returns: out: (batch_size, ...)
utils.py:146
↓ 7 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
models.py:2032
↓ 7 callersFunctioninterpolate
Interpolate data in time domain. This is used to compensate the resolution reduction in downsampling of a CNN. Args: x: (batch_siz
utils.py:158
↓ 6 callersMethodforward_features
(self, x)
model/htsat.py:577
↓ 4 callersMethod__init__
(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_
model/htsat.py:330
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
models.py:728
↓ 4 callersFunction_ntuple
(n)
model/layers.py:20
↓ 4 callersFunction_resnet_conv1x1
(in_planes, out_planes)
models.py:576
↓ 4 callersFunctioncreate_folder
(fd)
utils.py:81
↓ 4 callersFunctionint16_to_float32
(x)
utils.py:96
↓ 4 callersFunctionpad_framewise_output
Pad framewise_output to the same length as input frames. The pad value is the same as the value of the last frame. Args: framewise_out
utils.py:175
↓ 4 callersMethodreshape_wav2img
(self, x)
model/htsat.py:729
↓ 4 callersFunctiontrunc_normal_
r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi
model/layers.py:150
↓ 3 callersFunction_resnet_conv3x3
(in_planes, out_planes)
models.py:570
↓ 3 callersMethodevaluate_metric
(self, pred, ans)
sed_model.py:43
↓ 2 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/layers.py:97
↓ 2 callersFunction_resnet_conv1x1_wav1d
(in_planes, out_planes)
models.py:1939
↓ 2 callersFunction_resnet_conv3x1_wav1d
(in_planes, out_planes, dilation)
models.py:1933
↓ 2 callersMethodcrop_wav
(self, x, crop_size, spe_pos = None)
model/htsat.py:717
↓ 2 callersFunctiond_prime
(auc)
utils.py:128
↓ 2 callersFunctionfloat32_to_int16
(x)
utils.py:99
↓ 2 callersMethodgenerate_queue
(self)
data_generator.py:60
↓ 2 callersFunctionnorm_cdf
(x)
model/layers.py:117
↓ 2 callersFunctionprocess_idc
(index_path, classes_num, filename)
utils.py:105
↓ 2 callersMethodrepeat_wat2img
(self, x, cur_pos)
model/htsat.py:747
↓ 2 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
model/htsat.py:30
↓ 1 callersMethod__init__
(self, sed_model, config, dataset)
sed_model.py:36
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
model/layers.py:114
↓ 1 callersFunctioncombine_full_indexes
Combine all balanced and unbalanced indexes hdf5s to a single hdf5. This combined indexes hdf5 is used for training with full data (~20k balanced
create_indexes.py:40
↓ 1 callersFunctioncreate_indexes
Create indexes a for dataloader to read for training. When users have a new task and their own data, they need to create similar indexes. The
create_indexes.py:16
↓ 1 callersMethodcrop_wav
(self, x)
data_generator.py:91
↓ 1 callersFunctiondraw_timeline
(heatmap, f_class)
fl_evaluate.py:113
↓ 1 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). This is the same as the DropConnect impl I created for Ef
model/layers.py:34
↓ 1 callersFunctiondump_config
(config, filename, include_time = False)
utils.py:85
↓ 1 callersFunctionesm_test
()
main.py:104
↓ 1 callersMethodevaluate_metric
(self, pred, ans)
sed_model.py:267
↓ 1 callersFunctionfl_mapping
(heatmap, f_map)
fl_evaluate.py:98
↓ 1 callersMethodgenerate_queue
(self)
data_generator.py:204
↓ 1 callersMethodgenerate_queue
(self)
data_generator.py:244
↓ 1 callersFunctionget_loss_func
(loss_type)
utils.py:133
↓ 1 callersFunctionget_mix_lambda
(mixup_alpha, batch_size)
utils.py:77
↓ 1 callersMethodinit_weight
(self)
models.py:46
↓ 1 callersMethodinit_weight
(self)
models.py:86
↓ 1 callersMethodinit_weight
(self)
models.py:181
↓ 1 callersMethodinit_weight
(self)
models.py:268
↓ 1 callersMethodinit_weight
(self)
models.py:356
↓ 1 callersMethodinit_weight
(self)
models.py:439
↓ 1 callersMethodinit_weight
(self)
models.py:524
↓ 1 callersMethodinit_weight
(self)
models.py:1074
↓ 1 callersMethodinit_weight
(self)
models.py:1165
↓ 1 callersMethodinit_weight
(self)
models.py:1256
↓ 1 callersMethodinit_weight
(self)
models.py:1566
↓ 1 callersMethodinit_weight
(self)
models.py:1620
↓ 1 callersMethodinit_weight
(self)
models.py:1660
↓ 1 callersMethodinit_weight
(self)
models.py:1717
↓ 1 callersMethodinit_weight
(self)
models.py:1759
↓ 1 callersMethodinit_weight
(self)
models.py:1842
↓ 1 callersMethodinit_weight
(self)
models.py:1895
↓ 1 callersMethodinit_weight
(self)
models.py:2103
↓ 1 callersMethodinit_weight
(self)
models.py:2158
↓ 1 callersMethodinit_weight
(self)
models.py:2210
↓ 1 callersMethodinit_weight
(self)
models.py:2265
↓ 1 callersMethodinit_weight
(self)
models.py:2363
↓ 1 callersMethodinit_weight
(self)
models.py:2477
↓ 1 callersMethodinit_weight
(self)
models.py:2590
↓ 1 callersMethodinit_weight
(self)
models.py:2687
↓ 1 callersMethodinit_weight
(self)
models.py:2778
↓ 1 callersMethodinit_weight
(self)
models.py:2871
↓ 1 callersMethodinit_weight
(self)
models.py:2962
↓ 1 callersMethodinit_weight
(self)
models.py:3055
↓ 1 callersMethodinit_weight
(self)
models.py:3153
↓ 1 callersMethodinit_weight
(self)
models.py:3258
↓ 1 callersMethodinit_weights
(self)
models.py:121
↓ 1 callersMethodinit_weights
(self)
models.py:607
↓ 1 callersMethodinit_weights
(self)
models.py:662
↓ 1 callersMethodinit_weights
(self)
models.py:813
↓ 1 callersMethodinit_weights
(self)
models.py:900
↓ 1 callersMethodinit_weights
(self)
models.py:987
↓ 1 callersMethodinit_weights
(self)
models.py:1385
↓ 1 callersMethodinit_weights
(self)
models.py:1970
↓ 1 callersFunctionmain
()
fl_evaluate.py:131
↓ 1 callersFunctionmain
()
convert_desed.py:12
↓ 1 callersFunctionmain
()
main.py:401
↓ 1 callersMethodnonlinear_transform
(self, x)
models.py:133
↓ 1 callersFunctionprocess
(f_dir, f_name, f_class, f_map)
fl_evaluate.py:50
↓ 1 callersFunctionsave_idc
()
main.py:80
↓ 1 callersFunctiontest
()
main.py:173
↓ 1 callersMethodtime_shifting
(self, x)
data_generator.py:54
↓ 1 callersMethodtime_shifting
(self, x, shift_len)
sed_model.py:150
↓ 1 callersMethodtime_shifting
(self, x, shift_len)
sed_model.py:298
↓ 1 callersFunctiontrain
()
main.py:251
↓ 1 callersFunctionvariance_scaling_
(tensor, scale=1.0, mode='fan_in', distribution='normal')
model/layers.py:171
↓ 1 callersFunctionweight_average
()
main.py:86
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
model/htsat.py:44
Method__getitem__
Load waveform and target of an audio clip. Args: index: the index number Return: { "hdf5_path": str,
data_generator.py:96
Method__getitem__
Load waveform and target of an audio clip. Args: index: the index number Return: { "audio_name": str,
data_generator.py:209
Method__getitem__
Load waveform and target of an audio clip. Args: index: the index number Return: { "audio_name": str,
data_generator.py:249
next →1–100 of 248, ranked by callers