Merge pull request #60 from qingqing01/develop

bug fix for image_classification

image_classification/README.md

@@ -188,7 +188,9 @@ def process(settings, file_list):
             labels = batch['labels']
             for im, lab in zip(images, labels):
                 if settings.is_train and np.random.randint(2):
+                    im = im.reshape(3, 32, 32)
                     im = im[:,:,::-1]
+                    im = im.flatten()
                 im = im - settings.mean
                 yield {
                     'image': im.astype('float32'),

image_classification/dataprovider.py

@@ -38,6 +38,8 @@ def process(settings, file_list):
             labels = batch['labels']
             for im, lab in zip(images, labels):
                 if settings.is_train and np.random.randint(2):
+                    im = im.reshape(3, 32, 32)
                     im = im[:, :, ::-1]
+                    im = im.flatten()
                 im = im - settings.mean
                 yield {'image': im.astype('float32'), 'label': int(lab)}