赞
踩
其实忘了加空格:
将
frames = rearrange(frames, 'bhwc -> bchw')
改为:
frames = rearrange(frames, 'b h w c -> b c h w')
就好了