Merge pull request lisa-lab#72 from nouiz/swapaxes

lamblin · lamblin · commit 408ea682e1d7 · 2015-02-13T14:58:17.000-05:00
clarify a section.
diff --git a/doc/lenet.txt b/doc/lenet.txt
@@ -194,7 +194,12 @@ one of Figure 1. The input consists of 3 features maps (an RGB color image) of s
 
 .. code-block:: python
 
+        import theano
+        from theano import tensor as T
         from theano.tensor.nnet import conv
+
+        import numpy
+
         rng = numpy.random.RandomState(23455)
 
         # instantiate 4D tensor for input
@@ -261,10 +266,11 @@ Let's have a little bit of fun with this...
 
         # open random image of dimensions 639x516
         img = Image.open(open('doc/images/3wolfmoon.jpg'))
+        # dimensions are (height, width, channel)
         img = numpy.asarray(img, dtype='float64') / 256.
 
         # put image in 4D tensor of shape (1, 3, height, width)
-        img_ = img.swapaxes(0, 2).swapaxes(1, 2).reshape(1, 3, 639, 516)
+        img_ = img.transpose(2, 0, 1).reshape(1, 3, 639, 516)
         filtered_img = f(img_)
 
         # plot original image and first and second components of output
