Update codes.py

codes.py

@@ -152,7 +152,7 @@ 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