Update codes.py

codes.py

@@ -153,7 +153,7 @@ def wrap_points(img, x_source, y_source, l=1, DIM =dim):
         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
+            img[:,:,y0-l:y0+l,x0-l:x0+l] = 0
     return img