Update codes.py

codes.py

@@ -148,11 +148,11 @@ def transform_standard_points(Affine_mat, x,y):
     yt0 = XYt[1]
     return xt0, yt0
 
-def wrap_points(img, x_source, y_source, l=1):
+def wrap_points(img, x_source, y_source, l=1, DIM =128):
  for i in range(len(y_source)):
     x0 = x_source[i].int()
     y0 = y_source[i].int()
-    if (x0<dim) and (x0>0) and (y0<dim) and (y0>0):
+    if (x0<DIM) and (x0>0) and (y0<DIM) and (y0>0):
         img[:,:,x0-l:x0+l,y0-l:y0+l] = 0
     return img