Update codes.py

codes.py

@@ -148,12 +148,12 @@ def transform_standard_points(Affine_mat, x,y):
     yt0 = XYt[1]
     return xt0, yt0
 
-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):
-        img[:,:,x0-l:x0+l,y0-l:y0+l] = 0
+def wrap_points(img, x_source, y_source, l=1, DIM =dim):
+    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):
+            img[:,:,x0-l:x0+l,y0-l:y0+l] = 0
     return img