SYMBOL INDEX (181 symbols across 17 files)

FILE: beat_this/cli.py
  function get_parser (line 22) | def get_parser():
  function derive_output_path (line 92) | def derive_output_path(input_path, suffix, append, output=None, parent=N...
  function run (line 114) | def run(
  function main (line 194) | def main():

FILE: beat_this/dataset/augment.py
  function augment_pitchtempo (line 5) | def augment_pitchtempo(item, augmentations):
  function augment_pitch (line 42) | def augment_pitch(item, pitch_params):
  function augment_tempo (line 50) | def augment_tempo(item, tempo_params):
  function stretch_annotations (line 60) | def stretch_annotations(item, percentage):
  function shift_annotations (line 71) | def shift_annotations(item, semitones):
  function stretch_filename (line 76) | def stretch_filename(item, percentage):
  function shift_filename (line 85) | def shift_filename(item, semitones):
  function number_of_precomputed_augmentations (line 94) | def number_of_precomputed_augmentations(augmentations):
  function precomputed_augmentation_filenames (line 105) | def precomputed_augmentation_filenames(augmentations, ext="npy"):
  function augment_mask_ (line 129) | def augment_mask_(spect, augmentations: dict, fps: int):
  function apply_mask_excerpt (line 177) | def apply_mask_excerpt(excerpt, kind, min_parts, max_parts):

FILE: beat_this/dataset/dataset.py
  class BeatTrackingDataset (line 23) | class BeatTrackingDataset(Dataset):
    method __init__ (line 38) | def __init__(
    method _load_dataset_infos (line 81) | def _load_dataset_infos(self, datasets):
    method _load_spect_bundles (line 88) | def _load_spect_bundles(self, datasets):
    method _load_dataset_item (line 96) | def _load_dataset_item(self, item_name):
    method _get_spect (line 146) | def _get_spect(self, item):
    method get_frame_count (line 154) | def get_frame_count(self, index):
    method get_beat_count (line 158) | def get_beat_count(self, index):
    method get_downbeat_count (line 162) | def get_downbeat_count(self, index):
    method __len__ (line 166) | def __len__(self):
    method __getitem__ (line 169) | def __getitem__(self, index):
  class BeatDataModule (line 247) | class BeatDataModule(pl.LightningDataModule):
    method __init__ (line 268) | def __init__(
    method setup (line 305) | def setup(self, stage):
    method train_dataloader (line 448) | def train_dataloader(self):
    method val_dataloader (line 458) | def val_dataloader(self):
    method test_dataloader (line 465) | def test_dataloader(self):
    method predict_dataloader (line 468) | def predict_dataloader(self):
    method get_train_positive_weights (line 473) | def get_train_positive_weights(self, widen_target_mask=3):
  function prepare_annotations (line 512) | def prepare_annotations(item, start_frame, end_frame, fps):

FILE: beat_this/dataset/mmnpz.py
  class MemmappedNpzFile (line 12) | class MemmappedNpzFile(Mapping):
    method __init__ (line 38) | def __init__(self, fn: str, cache: bool = True, preload: bool = False):
    method load (line 54) | def load(self, name: str):
    method close (line 78) | def close(self):
    method __enter__ (line 83) | def __enter__(self):
    method __exit__ (line 86) | def __exit__(self, exc_type, exc_value, traceback):
    method __iter__ (line 89) | def __iter__(self):
    method __len__ (line 92) | def __len__(self):
    method __getitem__ (line 95) | def __getitem__(self, key: str):
    method __contains__ (line 106) | def __contains__(self, key: str):
  class MemoryviewIO (line 111) | class MemoryviewIO(object):
    method __init__ (line 116) | def __init__(self, buffer):
    method seek (line 123) | def seek(self, offset, whence=0):
    method read (line 131) | def read(self, size=-1):
    method tell (line 138) | def tell(self):

FILE: beat_this/inference.py
  function load_checkpoint (line 16) | def load_checkpoint(checkpoint_path: str, device: str | torch.device = "...
  function load_model (line 56) | def load_model(
  function zeropad (line 90) | def zeropad(spect: torch.Tensor, left: int = 0, right: int = 0):
  function split_piece (line 100) | def split_piece(
  function aggregate_prediction (line 138) | def aggregate_prediction(
  function split_predict_aggregate (line 188) | def split_predict_aggregate(
  class Spect2Frames (line 233) | class Spect2Frames:
    method __init__ (line 238) | def __init__(self, checkpoint_path="final0", device="cpu", float16=Fal...
    method spect2frames (line 244) | def spect2frames(self, spect):
    method __call__ (line 256) | def __call__(self, spect):
  class Audio2Frames (line 260) | class Audio2Frames(Spect2Frames):
    method __init__ (line 265) | def __init__(self, checkpoint_path="final0", device="cpu", float16=Fal...
    method signal2spect (line 269) | def signal2spect(self, signal, sr):
    method __call__ (line 279) | def __call__(self, signal, sr):
  class Audio2Beats (line 284) | class Audio2Beats(Audio2Frames):
    method __init__ (line 295) | def __init__(
    method __call__ (line 301) | def __call__(self, signal, sr):
  class File2Beats (line 306) | class File2Beats(Audio2Beats):
    method __call__ (line 307) | def __call__(self, audio_path):
  class File2File (line 312) | class File2File(File2Beats):
    method __call__ (line 313) | def __call__(self, audio_path, output_path):

FILE: beat_this/model/beat_tracker.py
  class BeatThis (line 18) | class BeatThis(nn.Module):
    method __init__ (line 38) | def __init__(
    method make_stem (line 109) | def make_stem(spect_dim: int, stem_dim: int) -> nn.Module:
    method make_frontend_block (line 129) | def make_frontend_block(
    method _init_weights (line 171) | def _init_weights(module: nn.Module):
    method forward (line 188) | def forward(self, x):
    method _load_from_state_dict (line 194) | def _load_from_state_dict(self, state_dict, prefix, *args, **kwargs):
    method state_dict (line 199) | def state_dict(self, *args, **kwargs):
  class PartialRoformer (line 206) | class PartialRoformer(nn.Module):
    method __init__ (line 213) | def __init__(
    method forward (line 238) | def forward(self, x):
  class PartialFTTransformer (line 251) | class PartialFTTransformer(nn.Module):
    method __init__ (line 259) | def __init__(
    method forward (line 290) | def forward(self, x):
  class SumHead (line 304) | class SumHead(nn.Module):
    method __init__ (line 311) | def __init__(self, input_dim):
    method forward (line 315) | def forward(self, x):
  class Head (line 333) | class Head(nn.Module):
    method __init__ (line 338) | def __init__(self, input_dim):
    method forward (line 342) | def forward(self, x):

FILE: beat_this/model/loss.py
  class MaskedBCELoss (line 9) | class MaskedBCELoss(torch.nn.Module):
    method __init__ (line 19) | def __init__(self, pos_weight: float = 1):
    method forward (line 27) | def forward(
  class ShiftTolerantBCELoss (line 38) | class ShiftTolerantBCELoss(torch.nn.Module):
    method __init__ (line 56) | def __init__(self, pos_weight: float = 1, tolerance: int = 3):
    method spread (line 65) | def spread(self, x: torch.Tensor, factor: int = 1):
    method crop (line 70) | def crop(self, x: torch.Tensor, factor: int = 1):
    method forward (line 73) | def forward(
  class SplittedShiftTolerantBCELoss (line 95) | class SplittedShiftTolerantBCELoss(torch.nn.Module):
    method __init__ (line 109) | def __init__(self, pos_weight: float = 1, tolerance: int = 3):
    method spread (line 120) | def spread(self, x: torch.Tensor, amount: int):
    method crop (line 126) | def crop(self, x: torch.Tensor, desired_length: int):
    method forward (line 135) | def forward(self, preds: torch.Tensor, targets: torch.Tensor, mask: to...

FILE: beat_this/model/pl_module.py
  class PLBeatThis (line 21) | class PLBeatThis(LightningModule):
    method __init__ (line 22) | def __init__(
    method _compute_loss (line 99) | def _compute_loss(self, batch, model_prediction):
    method _compute_metrics (line 116) | def _compute_metrics(self, batch, postp_beat, postp_downbeat, step="va...
    method _compute_metrics_target (line 132) | def _compute_metrics_target(self, batch, postp_target, target, step):
    method log_losses (line 164) | def log_losses(self, losses, batch_size, step="train"):
    method log_metrics (line 187) | def log_metrics(self, metrics, batch_size, step="val"):
    method training_step (line 199) | def training_step(self, batch, batch_idx):
    method validation_step (line 207) | def validation_step(self, batch, batch_idx):
    method test_step (line 224) | def test_step(self, batch, batch_idx):
    method predict_step (line 231) | def predict_step(
    method configure_optimizers (line 279) | def configure_optimizers(self):
    method _load_from_state_dict (line 308) | def _load_from_state_dict(self, state_dict, prefix, *args, **kwargs):
    method state_dict (line 313) | def state_dict(self, *args, **kwargs):
  class Metrics (line 320) | class Metrics:
    method __init__ (line 321) | def __init__(self, eval_trim_beats: int) -> None:
    method __call__ (line 324) | def __call__(self, truth, preds, step) -> Any:
  class CosineWarmupScheduler (line 342) | class CosineWarmupScheduler(torch.optim.lr_scheduler._LRScheduler):
    method __init__ (line 350) | def __init__(self, optimizer, warmup, max_iters, raise_last=0, raise_t...
    method get_lr (line 356) | def get_lr(self):
    method get_lr_factor (line 360) | def get_lr_factor(self, step):

FILE: beat_this/model/postprocessor.py
  class Postprocessor (line 9) | class Postprocessor:
    method __init__ (line 24) | def __init__(self, type: str = "minimal", fps: int = 50):
    method __call__ (line 39) | def __call__(
    method postp_minimal (line 85) | def postp_minimal(self, beat, downbeat, padding_mask):
    method _postp_minimal_item (line 113) | def _postp_minimal_item(self, padded_beat_peaks, padded_downbeat_peaks...
    method postp_dbn (line 138) | def postp_dbn(self, beat, downbeat, padding_mask):
    method _postp_dbn_item (line 153) | def _postp_dbn_item(self, padded_beat_prob, padded_downbeat_prob, mask):
  function deduplicate_peaks (line 176) | def deduplicate_peaks(peaks, width=1) -> np.ndarray:

FILE: beat_this/model/roformer.py
  function exists (line 15) | def exists(val):
  class RMSNorm (line 22) | class RMSNorm(Module):
    method __init__ (line 23) | def __init__(self, size, dim=-1):
    method forward (line 31) | def forward(self, x):
  class FeedForward (line 38) | class FeedForward(Module):
    method __init__ (line 39) | def __init__(
    method forward (line 60) | def forward(self, x):
  class Attend (line 67) | class Attend(nn.Module):
    method __init__ (line 68) | def __init__(self, dropout=0.0, scale=None):
    method forward (line 73) | def forward(self, q, k, v):
  class Attention (line 83) | class Attention(Module):
    method __init__ (line 84) | def __init__(
    method forward (line 114) | def forward(self, x):
  class Transformer (line 138) | class Transformer(Module):
    method __init__ (line 139) | def __init__(
    method forward (line 176) | def forward(self, x):

FILE: beat_this/preprocessing.py
  function load_audio (line 6) | def load_audio(path, dtype="float64"):
  class LogMelSpect (line 27) | class LogMelSpect(torch.nn.Module):
    method __init__ (line 28) | def __init__(
    method forward (line 56) | def forward(self, x):

FILE: beat_this/utils.py
  function index_to_framewise (line 7) | def index_to_framewise(index, length):
  function filename_to_augmentation (line 14) | def filename_to_augmentation(filename):
  function infer_beat_numbers (line 26) | def infer_beat_numbers(beats: np.ndarray, downbeats: np.ndarray) -> np.n...
  function save_beat_tsv (line 79) | def save_beat_tsv(beats: np.ndarray, downbeats: np.ndarray, outpath: str...
  function replace_state_dict_key (line 105) | def replace_state_dict_key(state_dict: dict, old: str, new: str):

FILE: launch_scripts/clean_checkpoints.py
  function main (line 7) | def main(args):

FILE: launch_scripts/compute_paper_metrics.py
  function main (line 17) | def main(args):
  function datamodule_setup (line 159) | def datamodule_setup(checkpoint, num_workers, datasplit):
  function plmodel_setup (line 174) | def plmodel_setup(checkpoint, eval_trim_beats, dbn, gpu):
  function compute_predictions (line 213) | def compute_predictions(model, trainer, predict_dataloader, return_preds...
  function write_predictions (line 228) | def write_predictions(fn, preds, piece):

FILE: launch_scripts/preprocess_audio.py
  function save_audio (line 24) | def save_audio(path, waveform, samplerate, resample_from=None):
  function save_spectrogram (line 37) | def save_spectrogram(path, spectrogram, dtype=np.float16):
  class SpectCreation (line 45) | class SpectCreation:
    method __init__ (line 46) | def __init__(self, pitch_shift, time_stretch, audio_sr, mel_args, verb...
    method create_spects (line 89) | def create_spects(self):
    method create_spect_piece (line 114) | def create_spect_piece(self, preprocessed_audio_folder, beat_path, dat...
  class AudioPreprocessing (line 161) | class AudioPreprocessing(object):
    method __init__ (line 162) | def __init__(
    method preprocess_audio (line 215) | def preprocess_audio(self):
    method process_audio_file (line 236) | def process_audio_file(self, dataset_name, audio_path):
  function augment_audio_file (line 332) | def augment_audio_file(
  function create_npz (line 383) | def create_npz(spect_dir, npz_file, augmentations, verbose):
  function ints (line 396) | def ints(value):
  function main (line 401) | def main(orig_audio_paths, pitch_shift, time_stretch, verbose):

FILE: launch_scripts/train.py
  function main (line 13) | def main(args):

FILE: tests/test_inference.py
  function test_File2Beat (line 10) | def test_File2Beat():
  function test_Audio2Frames (line 18) | def test_Audio2Frames():