-
Notifications
You must be signed in to change notification settings - Fork 0
/
imutils.py
38 lines (26 loc) · 812 Bytes
/
imutils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import numpy as np
import cv2
def translate(image, x, y):
M=np.float32([[1,0,x],[0,1,y]])
shifted=cv2.warpAffine(image,M,(image.shape[1],image.shape[0]))
return shifted
def rotate(image,angle,center=None,scale=1.0):
(h,w)=image.shape[:2]
if center is None:
center=(w/2,h/2)
M=cv2.getRotationMatrix2D(center,angle,scale)
rotated=cv2.warpAffine(image,M,(w,h))
return rotated
def resize(image,width=None,height=None,inter=cv2.INTER_AREA):
dim=None
(h,w)=image.shape[:2]
if width is None and height is None:
return image
if width is None:
r=height/float(h)
dim=(int(w*r),height)
else:
r=width/float(w)
dim=(width,int(h*r))
resized=cv2.resize(image,dim,interpolation=inter)
return resized