sczhou / ProPainter

[ICCV 2023] ProPainter: Improving Propagation and Transformer for Video Inpainting
https://shangchenzhou.com/projects/ProPainter/
Other
5.63k stars 665 forks source link

High GPU usage #32

Open USTAADCOM opened 1 year ago

USTAADCOM commented 1 year ago

Here are the arguments that I set and I have a 10GB GPU I just uploaded a 29-second video (804 frames) my GPU is gone OOM. Give me suggestions how I can make my GPU usage less. I already set these param after reading the guidelines given in repo for memory inference. parser.add_argument( "--resize_ratio", type=float, default=1.0, help='Resize scale for processing video.') parser.add_argument( '--height', type=int, default=240, help='Height of the processing video.') parser.add_argument( '--width', type=int, default=432, help='Width of the processing video.') parser.add_argument( '--mask_dilation', type=int, default=4, help='Mask dilation for video and flow masking.') parser.add_argument( "--ref_stride", type=int, default=20, help='Stride of global reference frames.') parser.add_argument( "--neighbor_length", type=int, default=5, help='Length of local neighboring frames.') parser.add_argument( "--subvideo_length", type=int, default=10, help='Length of sub-video for long video inference.')