espnet2.diar.label_processor.LabelProcessor

source

class espnet2.diar.label_processor.LabelProcessor(win_length: int = 512, hop_length: int = 128, center: bool = True)

Bases: Module

Label aggregator for speaker diarization

Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(input: Tensor, ilens: Tensor)

Forward.

• Parameters:
  • input – (Batch, Nsamples, Label_dim)
  • ilens – (Batch)
• Returns: (Batch, Frames, Label_dim) olens: (Batch)
• Return type: output