-
Notifications
You must be signed in to change notification settings - Fork 0
/
opencv.py
38 lines (30 loc) · 1.19 KB
/
opencv.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 cv2
import glob
import random
import os
import numpy as np
def createFolder(path):
os.makedirs(path)
def flipAndRotate(minangle, maxangle, flip,) :
for img in glob.glob("*.jpg") :
imagename = os.path.splitext(img)[0]
img = cv2.imread(img, 1)
createFolder(imagename)
for i in range(0, 5):
rows, cols ,zdim = img.shape
angle = random.randint(minangle, maxangle)
# cv2.getRotationMatrix2D(center, angle, scale)
M = cv2.getRotationMatrix2D((cols / 2, rows / 2), angle, 1)
# wrapAlline(source,martrix,size)
#verticleflip = 1, horixontalflip = 0
dst = cv2.warpAffine(img, M, (cols, rows))
resultimage = cv2.flip(dst, flip)
resultimage = np.array(resultimage)
resultimage[resultimage == 0] = 255
resultpath = imagename + "/"+imagename + str(i) + ".jpeg"
cv2.imwrite(resultpath, resultimage)
def takeInputArguments () :
print "arguments: minangle ,maxangle, flip "
print "verticleflip = 1, horixontalflip = 0"
flipAndRotate(int(input('minangle')), int(input('maxangle')), int(input('flip')))
takeInputArguments()