yanivnik/sinfusion-code

Small bug in main.py

Closed this issue · 1 comments

CaelynCheung1996 commented

Line 63 in main.py, crop_size = (int(frames[0].shape[-2] * 0.95), int(frames[0].shape[-1:] * 0.95)) should be changed to crop_size = (int(frames[0].shape[-2] * 0.95), int(frames[0].shape[-1] * 0.95)).

yanivnik commented

Thanks for noticing! Fixed.