Method push

scenedetect/detectors/transnet_v2.py:82–128

(self, pixels: np.ndarray, time: np.ndarray)

        return cuts

    def push(self, pixels: np.ndarray, time: np.ndarray):
        if self.pixels is None:
            self.pixels = pixels
            self.time = time

            return self._inference(
                np.stack(
                    (
                        np.tile(np.expand_dims(pixels[0], axis=0), (100, 1, 1, 1)),
                        np.concatenate(
                            (
                                np.tile(np.expand_dims(pixels[0], axis=0), (25, 1, 1, 1)),
                                pixels[:75],
                            ),
                            0,
                        ),
                    )
                ),
                np.stack(
                    (
                        np.tile(np.expand_dims(time[0], axis=0), (100,)),
                        np.concatenate(
                            (np.tile(np.expand_dims(time[0], axis=0), (25,)), time[:75]), 0
                        ),
                    )
                ),
            )
        else:
            # `self.time` is set in lockstep with `self.pixels` above, so it is non-None here.
            assert self.time is not None
            c1 = self.pixels
            c2 = pixels

            t1 = self.time
            t2 = time

            self.pixels = pixels
            self.time = time

            return self._inference(
                np.stack(
                    (np.concatenate((c1[25:], c2[:25]), 0), np.concatenate((c1[75:], c2[:75]), 0))
                ),
                np.stack(
                    (np.concatenate((t1[25:], t2[:25]), 0), np.concatenate((t1[75:], t2[:75]), 0))
                ),
            )


class TransnetV2Detector(SceneDetector):