From e6fac9ad6a30a997f05f17ce3bc279ecd35da2bd Mon Sep 17 00:00:00 2001 From: Newmu Date: Fri, 12 Sep 2014 14:48:06 -0400 Subject: [PATCH] Fixed uint8 all zeros bug --- indicoio/utils/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/indicoio/utils/__init__.py b/indicoio/utils/__init__.py index 12c83ab..4da1e26 100644 --- a/indicoio/utils/__init__.py +++ b/indicoio/utils/__init__.py @@ -95,13 +95,14 @@ def image_preprocess(image): image = np.asarray(image) if type(image).__module__ != np.__name__: raise ValueError('Image was not of type numpy.ndarray or list.') - if image.max() > 1: + if str(image.dtype) in ['int64','uint8']: image = image/255. if len(image.shape) == 2: image = np.dstack((image,image,image)) if len(image.shape) == 4: image = image[:,:,:3] - + print image + print image.dtype,image.min(),image.max() image = resize(image,(64,64)) image = image.tolist() return image \ No newline at end of file