o Sik@srddlZddlmZddlmZddlmZmZmZGdddZ de de d e fd d Z de de d e fd d ZdS)N)Optional)CutSet) LOG_EPSILONSecondsexactly_one_not_nullc@sheZdZdZdddedddfdeedeedeeded e d e d e d dfd dZ de d e fddZ dS) ExtraPaddinga0 A transform on batch of cuts (``CutSet``) that adds a number of extra context frames/samples/seconds on both sides of the cut. Exactly one type of duration has to specified in the constructor. It is intended mainly for training frame-synchronous ASR models with convolutional layers to avoid using padding inside of the hidden layers, by giving the model larger context in the input. Another useful application is to shift the input by a little, so that the data seen after frame subsampling is a bit different, which makes this a data augmentation technique. This is best used as the first transform in the transform list for dataset - it will ensure that each individual cut gets extra context before concatenation, or that it will be filled with noise, etc. NFboth extra_frames extra_samples extra_secondspad_feat_value randomized preserve_id directionreturncCsRt|||s Jd|dvsJd||_||_||_||_||_||_||_dS)a ExtraPadding's constructor. :param extra_frames: The total number of frames to add to each cut. We will add half that number on each side of the cut ("both" directions padding). :param extra_samples: The total number of samples to add to each cut. We will add half that number on each side of the cut ("both" directions padding). :param extra_seconds: The total duration in seconds to add to each cut. We will add half that number on each side of the cut ("both" directions padding). :param pad_feat_value: When padding a cut with precomputed features, what value should be used for padding (the default is a very low log-energy). :param randomized: When ``True``, we will sample a value from a uniform distribution of ``[0, extra_X]`` for each cut (for samples/frames -- sample an int, for duration -- sample a float). :param preserve_id: When ``True``, preserves the IDs the cuts had before augmentation. Otherwise, new random IDs are generated for the augmented cuts (default). :param direction: The padding direction. zSFor ExtraPadding, you have to specify exactly one of: frames, samples, or duration.)rleftrightz&Only three padding modes are supportedN)rr r r r r rr)selfr r r r r rrr_/home/ubuntu/.local/lib/python3.10/site-packages/lhotse/dataset/cut_transforms/extra_padding.py__init__s  zExtraPadding.__init__cutscsnjdurtfdd|DSjdur"tfdd|DSjdur3tfdd|DStd)Nc3:|]}|j|jtjjdjjjdVqdS)valuesample) num_framesr rrN)padrmaybe_sample_intr r r rr.0crrr Hs z(ExtraPadding.__call__..c3s6|]}|j|jtjjdjjdVqdS)r) num_samplesrrN)rr$rr r rrrr"rrr#Ss  c3r)r)durationr rrN)rr%maybe_sample_floatr r r rrrr"rrr#_s  z@Implementation error in ExtraPadding (please report this issue).)r r from_cutsr r ValueError)rrrr"r__call__Fs      zExtraPadding.__call__)__name__ __module__ __qualname____doc__rrintrfloatboolstrrrr)rrrrrs6 ,rrrrcC|rtd|}|SNr)randomrandintrrrrrq rcCr2r3)r4uniformrrrrr&wr6r&)r4typingrlhotser lhotse.utilsrrrrr.r0rr/r&rrrrs  i