Utilize numpy functions to do the color correction

mahmoudnafifi / WB_sRGB

White balance camera-rendered sRGB images (CVPR 2019) [Matlab & Python]

Other

344 stars 62 forks source link

Utilize numpy functions to do the color correction #8

Closed manipopopo closed 3 years ago

manipopopo commented 3 years ago

The PR utilizes broadcasting mechanism and numpy functions to speed up the color correction.

On a machine with 8-core 3.1 Ghz CPU, the original correctImage takes ~1.88 second to process example_images/figure3.jpg. With the PR, it takes ~0.14 second to process the same image.