Lornatang / SRGAN-PyTorch

A simple and complete implementation of super-resolution paper.
Apache License 2.0
421 stars 104 forks source link

Update test_image.py #9

Closed nisargshah1999 closed 3 years ago

nisargshah1999 commented 3 years ago

Update code for calculating inference time and FPS

Lornatang commented 3 years ago

This does not reflect the speed of GPU, and it doesn't make sense to do so.

nisargshah1999 commented 3 years ago

This is not to calculate the speed of the model, it is to calculate the inference speed of model in FPS and sec. for particular GPU

Lornatang commented 3 years ago

To be sure, in a single sample, the inference speed of GPU model is similar to that of CPU.

Lornatang commented 3 years ago

@nisargshah1999 Sorry, I can't merge this right now.