one-liner for src array

AllenInstitute · Jul 26, 2019 · 3961d07 · 3961d07
1 parent 945168c
commit 3961d07
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/renderapi/transform/leaf/thin_plate_spline.py b/renderapi/transform/leaf/thin_plate_spline.py
@@ -435,10 +435,7 @@ def src_array(xmin, ymin, xmax, ymax, nx, ny):
             (nx * ny) x 2 array of coordinated.
 
         """
-        x = np.linspace(xmin, xmax, nx)
-        y = np.linspace(ymin, ymax, ny)
-        xt, yt = np.meshgrid(x, y)
-        src = np.vstack((xt.flatten(), yt.flatten())).transpose()
+        src = np.mgrid[xmin:xmax:nx*1j, ymin:ymax:ny*1j].reshape(2, -1).T
         return src
 
     def scale_coordinates(