wassname/IndicoIo-python
1
0
Fork 0
mirror of https://github.com/wassname/IndicoIo-python.git synced 2026-08-09 11:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed uint8 all zeros bug

Browse Source
This commit is contained in:
Newmu
2014-09-12 14:48:06 -04:00
parent d6cb1bf1d9
commit e6fac9ad6a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
indicoio/utils/__init__.py
+3 -2
View File
@@ -95,13 +95,14 @@ def image_preprocess(image):
image = np.asarray(image) image = np.asarray(image)
if type(image).__module__ != np.__name__: if type(image).__module__ != np.__name__:
raise ValueError('Image was not of type numpy.ndarray or list.') 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. image = image/255.
if len(image.shape) == 2: if len(image.shape) == 2:
image = np.dstack((image,image,image)) image = np.dstack((image,image,image))
if len(image.shape) == 4: if len(image.shape) == 4:
image = image[:,:,:3] image = image[:,:,:3]
print image
print image.dtype,image.min(),image.max()
image = resize(image,(64,64)) image = resize(image,(64,64))
image = image.tolist() image = image.tolist()
return image return image