Method __init__

dali/test/python/operator_2/test_spectrogram.py:35–63

(
        self,
        device,
        batch_size,
        iterator,
        nfft,
        window_length,
        window_step,
        window=None,
        center=None,
        num_threads=1,
        device_id=0,
    )

class SpectrogramPipeline(Pipeline):
    def __init__(
        self,
        device,
        batch_size,
        iterator,
        nfft,
        window_length,
        window_step,
        window=None,
        center=None,
        num_threads=1,
        device_id=0,
    ):
        super(SpectrogramPipeline, self).__init__(batch_size, num_threads, device_id)
        self.device = device
        self.iterator = iterator
        self.inputs = ops.ExternalSource()
        window_fn = window(window_length).tolist() if window is not None else None
        self.fft = ops.Spectrogram(
            device=self.device,
            nfft=nfft,
            window_length=window_length,
            window_step=window_step,
            window_fn=window_fn,
            center_windows=center,
            power=2,
        )
        # randomly insert extra axis (channels?)
        self.r = np.random.randint(-1, 2)

    def define_graph(self):
        self.data = self.inputs()