Spaces:
Running
on
A10G
Running
on
A10G
resize large video to avoid out of memory.
Browse files
web-demos/hugging_face/inpainter/base_inpainter.py
CHANGED
@@ -205,6 +205,12 @@ class ProInpainter:
|
|
205 |
# The ouput size should be divided by 2 so that it can encoded by libx264
|
206 |
size = (int(ratio*size[0])//2*2, int(ratio*size[1])//2*2)
|
207 |
|
|
|
|
|
|
|
|
|
|
|
|
|
208 |
frames_len = len(frames)
|
209 |
frames, size, out_size = resize_frames(frames, size)
|
210 |
flow_masks, masks_dilated = read_mask_demo(masks, frames_len, size, dilate_radius, dilate_radius)
|
|
|
205 |
# The ouput size should be divided by 2 so that it can encoded by libx264
|
206 |
size = (int(ratio*size[0])//2*2, int(ratio*size[1])//2*2)
|
207 |
|
208 |
+
# TODO: hard code to reduce memory
|
209 |
+
if max(size[0], size[1]) > 720:
|
210 |
+
scale = 720.0 / max(size[0], size[1])
|
211 |
+
# The ouput size should be divided by 2 so that it can encoded by libx264
|
212 |
+
size = (int(scale*size[0])//2*2, int(scale*size[1])//2*2)
|
213 |
+
|
214 |
frames_len = len(frames)
|
215 |
frames, size, out_size = resize_frames(frames, size)
|
216 |
flow_masks, masks_dilated = read_mask_demo(masks, frames_len, size, dilate_radius, dilate_radius)
|