l

Signed-off-by: xujinling <jinling.xu@zilliz.com>
3 years ago · 9152c80f39
1 changed files with 2 additions and 1 deletions
--- a/bridge_former.py
+++ b/bridge_former.py
@ -61,7 +61,8 @@ class BridgeFormer(NNOperator):
        assert len(video.shape) == 4
        video = video.transpose(3, 0, 1, 2)  # thwc -> cthw
        if self.skip_preprocess:
            self.transform_cfgs.update(num_frames=None)
        video = transform_video(
            video=video,
            **self.transform_cfgs