search

zsyzzsoft / co-mod-gan

[ICLR 2021, Spotlight] Large Scale Image Completion via Co-Modulated Generative Adversarial Networks
Other
445 stars 67 forks source link

The time used to test a image #12

Closed yezhanglang closed 3 years ago

yezhanglang commented 3 years ago

This is a really great job! However, when I run the run_generator.py script, It takes more than 20s to test a image. Is that normal?

yezhanglang commented 3 years ago

By the way, I use png image to test. And my gpu has 9G video memory.

yezhanglang commented 3 years ago

And I modify the run_generator.py script to test a directory of images, after test some images, it keeps OOM.

zsyzzsoft commented 3 years ago

The script includes TensorFlow initialization, network loading and compiling, which in all take quite long. If you have multiple images, you should only include these lines in a for loop.

yezhanglang commented 3 years ago

I see, it's my fault! I haven't check it carefully. Thank you so much! 感谢大佬！