The size of tensor a (200) must match the size of tensor b (201) at non-singleton dimension 1

sczhou / ProPainter

[ICCV 2023] ProPainter: Improving Propagation and Transformer for Video Inpainting

https://shangchenzhou.com/projects/ProPainter/

Other

5.45k stars 645 forks source link

The size of tensor a (200) must match the size of tensor b (201) at non-singleton dimension 1 #84

Open rospin opened 3 months ago

rospin commented 3 months ago

I encountered the following problem while processing a .mp4 file

Processing: XXX_seg [200 frames]... Traceback (most recent call last): File "D:\Users\ProPainter\inference_propainter.py", line 372, in masked_frames = frames * (1 - masks_dilated) RuntimeError: The size of tensor a (200) must match the size of tensor b (201) at non-singleton dimension 1