:j ddlmZ ddefdZ ddedeedeedeed eedef d Z dd Z dd edeedeedeed eedef d Z ddefdZ ddedeedeedeed eedef dZy))Listreturnc6d|d|zz||dz zz dz |zzS)aCompute expected number of frames after 1D convolution Parameters ---------- num_samples : int Number of samples in the input signal kernel_size : int Kernel size stride : int Stride padding : int Padding dilation : int Dilation Returns ------- num_frames : int Number of frames in the output signal Source ------ https://pytorch.org/docs/stable/generated/torch.nn.Conv1d.html#torch.nn.Conv1d ) num_samples kernel_sizestridepaddingdilations y/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/pyannote/audio/utils/receptive_field.pyconv1d_num_framesrs16 a'k)H a,HH1LQWW WWNr r r r r cZ|}t||||D]\}}}} t||||| }|S)N)r r r r )zipr) r r r r r num_frameskspds rmulti_conv_num_framesr8sHJ+vwA 1a& AaQ  rc:d|dz |zz}||dz |zzd|zz S)aeCompute size of receptive field Parameters ---------- num_frames : int, optional Number of frames in the output signal kernel_size : int Kernel size stride : int Stride padding : int Padding dilation : int Dilation Returns ------- size : int Receptive field size rrr)rr r r r effective_kernel_sizes rconv1d_receptive_field_sizerHs50qH << JNf#< 2Q61< ( "   "!r)rrr)NNNN)rr'rrr)rr'rrr) typingrintrrrr r#r&rrrr*sR.?@XX@"  c  I #Y  3i   "@AK<"   c  I #Y  3i    *;<GG<" " "c" I"#Y " 3i "  "r