search

vt-vl-lab / FGVC

[ECCV 2020] Flow-edge Guided Video Completion
Other
1.55k stars 263 forks source link

Near zero GPU load #34

Closed coldfir3 closed 3 years ago

coldfir3 commented 3 years ago

I am having almost zero % GPU load when running the same code from colab. Any ideas what might be causing it? f

brunomsantiago commented 3 years ago

The gpu intensive neural networks are just part of the algorithm. The video completion code has lots of cpu-intensive operations using numpy, scipy, skimage, opencv and pillow. Can't say for sure, but I belive the video completion tool spend most of time on cpu operations.

gaochen315 commented 3 years ago

Thanks, @brunomsantiago. You are right. Our method uses GPU to predict the flow, complete the flow edge, and perform per-frame inpainting. But the majority of the code is about content propagation, which is pure CPU work.