improc ass4: Added Gauss diff 3d plot.

improc/ass4/gauss.py

@@ -59,7 +59,8 @@ elif argv[1] == 'diff':
     if len(argv) < 3:
         exit_with_usage()
 
-    W = Gauss(int(argv[2]))
+    s = int(argv[2])
+    W = Gauss(s)
     G = convolve(F, W, mode='nearest')
 
     # Original image
@@ -70,7 +71,8 @@ elif argv[1] == 'diff':
     x = arange(W.shape[0])
     X, Y = meshgrid(x, x)
     ax = subplot(132, projection='3d')
-    ax.plot_surface(X, Y, W, rstride=1, cstride=1, cmap='jet')
+    stride = s / 3
+    ax.plot_surface(X, Y, W, rstride=stride, cstride=stride, cmap='jet')
     ax.set_xlabel('x')
     ax.set_ylabel('y')
     ax.set_zlabel('g(x, y)')