Commit ef117f9d authored by Yuxin Wu's avatar Yuxin Wu

fix syntax error

parent d46f15b5
......@@ -25,6 +25,6 @@ class Rotation(ImageAugmentor):
return cv2.getRotationMatrix2D(tuple(center), deg, 1)
def _augment(self, img, rot_m):
return cv2.warpAffine(img rot_m, img.shape[1::-1],
return cv2.warpAffine(img, rot_m, img.shape[1::-1],
flags=cv2.INTER_CUBIC, borderMode=cv2.BORDER_REPLICATE)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment