Added test case LightCNN9

bob/learn/tensorflow/layers/Maxout.py

@@ -94,6 +94,11 @@ class MaxOut(base.Layer):
         for i in range(len(shape)):
             if shape[i] is None:
                 shape[i] = gen_array_ops.shape(inputs)[i]
+
         outputs = math_ops.reduce_max(gen_array_ops.reshape(inputs, shape), -1, keep_dims=False)
+        shape = outputs.get_shape().as_list()
+        shape[self.axis] = self.num_units
+        outputs.set_shape(shape)
+
 
         return outputs

bob/learn/tensorflow/network/LightCNN9.py

@@ -144,12 +144,12 @@ class LightCNN9(object):
                                          reuse=reuse)
             graph = maxout(graph,
                            num_units=256,
-                           scope='Maxoutfc1')
+                           name='Maxoutfc1')
 
             graph = slim.fully_connected(graph, self.n_classes,
                                          weights_initializer=initializer,
                                          activation_fn=None,
-                                         scope='fc1',
+                                         scope='fc2',
                                          reuse=reuse)
 
         return graph

bob/learn/tensorflow/test/test_cnn.py

@@ -123,7 +123,7 @@ def test_cnn_trainer():
     del trainer
     del graph
 
-"""
+
 def test_lightcnn_trainer():
 
     # generating fake data
@@ -158,7 +158,7 @@ def test_lightcnn_trainer():
 
     # One graph trainer
     trainer = Trainer(train_data_shuffler,
-                      iterations=50,
+                      iterations=5,
                       analizer=None,
                       temp_dir=directory
                       )
@@ -178,7 +178,7 @@ def test_lightcnn_trainer():
     shutil.rmtree(directory)
     del trainer
     del graph
-"""
+