Update codes.py

codes.py

@@ -150,10 +150,11 @@ def transform_standard_points(Affine_mat, x,y):
 
 def wrap_points(img, x_source, y_source, l=1):
  for i in range(len(y_source)):
-   x0 = x_source[i].int()
-   y0 = y_source[i].int()
-   img[:,:,x0-l:x0+l,y0-l:y0+l] = 0
- return img
+    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
 
 
 def wrap_imge_cropped(Affine_mtrx, source_img, dim1=224, dim2=128):