search

mahmoudnafifi / WB_sRGB

White balance camera-rendered sRGB images (CVPR 2019) [Matlab & Python]
Other
344 stars 62 forks source link

processing speed of python version #6

Closed jahwanoh closed 4 years ago

jahwanoh commented 4 years ago

Hi, First of all, thank you for sharing this code. I tried with python version with high resolution video (4096x554) and it performs very slowly (4 fps). Processing speed for the video is not a target of this paper but is there any suggestion how to improve processing speed? I will try to search for how to use gpu with matrix operation because it seems numpy operations took much of the time.

mahmoudnafifi commented 4 years ago

Hi,

You can check this PR with an optimized Python version. It says it works 20X faster. I will review and merge it into the master branch soon, but just in case you are in a hurry.

jahwanoh commented 4 years ago

wow it is almost same approach but seems work well already.

Thank you!