-
Notifications
You must be signed in to change notification settings - Fork 0
/
Rotate.py
40 lines (36 loc) · 1.21 KB
/
Rotate.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
39
40
import numpy as np
import cv2
name= input('input name of image:')
rot=input('input the desired rotation angle (90, 180, or 270) or mirror:')
img=cv2.imread(name)
C=(int(len(img[0])))
R=(int(len(img)))
if rot=='90' or rot=='180' or rot=='270' or rot=='mirror':
if rot=='90':
editimg=np.zeros((C,R,3), dtype=int)
for j in range(R):
for i in range(C-1,-1,-1):
editimg[i][j]=img[j][-i]
if rot=='180':
editimg=np.zeros((R,C,3), dtype=int)
for i in range(R):
for j in range(C):
editimg[i, C-1-j] = img[R-1-i, j]
if rot=='270':
editimg=np.zeros((R,C,3), dtype=int)
for i in range(R):
for j in range(C):
editimg[i, C-1-j] = img[R-1-i, j]
for j in range(R):
for i in range(C-1,-1,-1):
editimg[i][j]=img[j][-i]
if rot=='mirror':
editimg=np.zeros((R,C,3), dtype=int)
for i in range(R):
for j in range(C):
editimg[i][j]=img[i][-j]
else: print('error: invalid input. Enter a 90,180,270, or mirror')
editimg=editimg.astype(np.uint8)
cv2.imshow('image', editimg)
cv2.waitKey(0)
cv2.destroyAllWindows()