From 57632b837b9b4eebdf710356dc2d534c3c87f1ad Mon Sep 17 00:00:00 2001 From: gamer13 Date: Thu, 30 Jul 2015 12:19:57 +0200 Subject: [PATCH] Update color_space.py --- color_space.py | 1 + 1 file changed, 1 insertion(+) diff --git a/color_space.py b/color_space.py index 1ce1cb0..3ade8ef 100644 --- a/color_space.py +++ b/color_space.py @@ -40,6 +40,7 @@ def to_HSV(I): def to_nRGB(I): _I = I / 255.0 norm_I = numpy.sqrt(_I[:, :, 0] ** 2 + _I[:, :, 1] ** 2 + _I[:, :, 2] ** 2) + norm_I[norm_I == 0] = 1 / numpy.sqrt(3) # To avoid division by 0 below, set these pixels to normalized gray norm_r = (_I[:, :, 0] / norm_I * 255).astype(numpy.uint8) norm_g = (_I[:, :, 1] / norm_I * 255).astype(numpy.uint8) norm_b = (_I[:, :, 2] / norm_I * 255).astype(numpy.uint8)