Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/RetroCirce/HTS-Audio-Transformer
/ functions
Functions
248 in github.com/RetroCirce/HTS-Audio-Transformer
⨍
Functions
248
◇
Types & classes
59
↓ 99 callers
Function
init_layer
Initialize a Linear or Convolutional layer.
models.py:11
↓ 65 callers
Function
init_bn
Initialize a Batchnorm layer.
models.py:20
↓ 41 callers
Method
__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:401
↓ 32 callers
Function
do_mixup
Args: x: (batch_size , ...) mixup_lambda: (batch_size,) Returns: out: (batch_size, ...)
utils.py:146
↓ 7 callers
Method
_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
models.py:2032
↓ 7 callers
Function
interpolate
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 callers
Method
forward_features
(self, x)
model/htsat.py:577
↓ 4 callers
Method
__init__
(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_
model/htsat.py:330
↓ 4 callers
Method
_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
models.py:728
↓ 4 callers
Function
_ntuple
(n)
model/layers.py:20
↓ 4 callers
Function
_resnet_conv1x1
(in_planes, out_planes)
models.py:576
↓ 4 callers
Function
create_folder
(fd)
utils.py:81
↓ 4 callers
Function
int16_to_float32
(x)
utils.py:96
↓ 4 callers
Function
pad_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 callers
Method
reshape_wav2img
(self, x)
model/htsat.py:729
↓ 4 callers
Function
trunc_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 callers
Function
_resnet_conv3x3
(in_planes, out_planes)
models.py:570
↓ 3 callers
Method
evaluate_metric
(self, pred, ans)
sed_model.py:43
↓ 2 callers
Method
__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/layers.py:97
↓ 2 callers
Function
_resnet_conv1x1_wav1d
(in_planes, out_planes)
models.py:1939
↓ 2 callers
Function
_resnet_conv3x1_wav1d
(in_planes, out_planes, dilation)
models.py:1933
↓ 2 callers
Method
crop_wav
(self, x, crop_size, spe_pos = None)
model/htsat.py:717
↓ 2 callers
Function
d_prime
(auc)
utils.py:128
↓ 2 callers
Function
float32_to_int16
(x)
utils.py:99
↓ 2 callers
Method
generate_queue
(self)
data_generator.py:60
↓ 2 callers
Function
norm_cdf
(x)
model/layers.py:117
↓ 2 callers
Function
process_idc
(index_path, classes_num, filename)
utils.py:105
↓ 2 callers
Method
repeat_wat2img
(self, x, cur_pos)
model/htsat.py:747
↓ 2 callers
Function
window_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 callers
Method
__init__
(self, sed_model, config, dataset)
sed_model.py:36
↓ 1 callers
Function
_no_grad_trunc_normal_
(tensor, mean, std, a, b)
model/layers.py:114
↓ 1 callers
Function
combine_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 callers
Function
create_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 callers
Method
crop_wav
(self, x)
data_generator.py:91
↓ 1 callers
Function
draw_timeline
(heatmap, f_class)
fl_evaluate.py:113
↓ 1 callers
Function
drop_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 callers
Function
dump_config
(config, filename, include_time = False)
utils.py:85
↓ 1 callers
Function
esm_test
()
main.py:104
↓ 1 callers
Method
evaluate_metric
(self, pred, ans)
sed_model.py:267
↓ 1 callers
Function
fl_mapping
(heatmap, f_map)
fl_evaluate.py:98
↓ 1 callers
Method
generate_queue
(self)
data_generator.py:204
↓ 1 callers
Method
generate_queue
(self)
data_generator.py:244
↓ 1 callers
Function
get_loss_func
(loss_type)
utils.py:133
↓ 1 callers
Function
get_mix_lambda
(mixup_alpha, batch_size)
utils.py:77
↓ 1 callers
Method
init_weight
(self)
models.py:46
↓ 1 callers
Method
init_weight
(self)
models.py:86
↓ 1 callers
Method
init_weight
(self)
models.py:181
↓ 1 callers
Method
init_weight
(self)
models.py:268
↓ 1 callers
Method
init_weight
(self)
models.py:356
↓ 1 callers
Method
init_weight
(self)
models.py:439
↓ 1 callers
Method
init_weight
(self)
models.py:524
↓ 1 callers
Method
init_weight
(self)
models.py:1074
↓ 1 callers
Method
init_weight
(self)
models.py:1165
↓ 1 callers
Method
init_weight
(self)
models.py:1256
↓ 1 callers
Method
init_weight
(self)
models.py:1566
↓ 1 callers
Method
init_weight
(self)
models.py:1620
↓ 1 callers
Method
init_weight
(self)
models.py:1660
↓ 1 callers
Method
init_weight
(self)
models.py:1717
↓ 1 callers
Method
init_weight
(self)
models.py:1759
↓ 1 callers
Method
init_weight
(self)
models.py:1842
↓ 1 callers
Method
init_weight
(self)
models.py:1895
↓ 1 callers
Method
init_weight
(self)
models.py:2103
↓ 1 callers
Method
init_weight
(self)
models.py:2158
↓ 1 callers
Method
init_weight
(self)
models.py:2210
↓ 1 callers
Method
init_weight
(self)
models.py:2265
↓ 1 callers
Method
init_weight
(self)
models.py:2363
↓ 1 callers
Method
init_weight
(self)
models.py:2477
↓ 1 callers
Method
init_weight
(self)
models.py:2590
↓ 1 callers
Method
init_weight
(self)
models.py:2687
↓ 1 callers
Method
init_weight
(self)
models.py:2778
↓ 1 callers
Method
init_weight
(self)
models.py:2871
↓ 1 callers
Method
init_weight
(self)
models.py:2962
↓ 1 callers
Method
init_weight
(self)
models.py:3055
↓ 1 callers
Method
init_weight
(self)
models.py:3153
↓ 1 callers
Method
init_weight
(self)
models.py:3258
↓ 1 callers
Method
init_weights
(self)
models.py:121
↓ 1 callers
Method
init_weights
(self)
models.py:607
↓ 1 callers
Method
init_weights
(self)
models.py:662
↓ 1 callers
Method
init_weights
(self)
models.py:813
↓ 1 callers
Method
init_weights
(self)
models.py:900
↓ 1 callers
Method
init_weights
(self)
models.py:987
↓ 1 callers
Method
init_weights
(self)
models.py:1385
↓ 1 callers
Method
init_weights
(self)
models.py:1970
↓ 1 callers
Function
main
()
fl_evaluate.py:131
↓ 1 callers
Function
main
()
convert_desed.py:12
↓ 1 callers
Function
main
()
main.py:401
↓ 1 callers
Method
nonlinear_transform
(self, x)
models.py:133
↓ 1 callers
Function
process
(f_dir, f_name, f_class, f_map)
fl_evaluate.py:50
↓ 1 callers
Function
save_idc
()
main.py:80
↓ 1 callers
Function
test
()
main.py:173
↓ 1 callers
Method
time_shifting
(self, x)
data_generator.py:54
↓ 1 callers
Method
time_shifting
(self, x, shift_len)
sed_model.py:150
↓ 1 callers
Method
time_shifting
(self, x, shift_len)
sed_model.py:298
↓ 1 callers
Function
train
()
main.py:251
↓ 1 callers
Function
variance_scaling_
(tensor, scale=1.0, mode='fan_in', distribution='normal')
model/layers.py:171
↓ 1 callers
Function
weight_average
()
main.py:86
↓ 1 callers
Function
window_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