77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
# @Time : 2022-7-25 0025 10:46
|
|||
|
|
# @Author : Qing
|
|||
|
|
# @Email : derighoid@gmail.com
|
|||
|
|
# @File : Orientation_Modification.py
|
|||
|
|
# @Software: PyCharm
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
# 导入库
|
|||
|
|
import matplotlib.pyplot as plt
|
|||
|
|
import numpy as np
|
|||
|
|
from PIL import Image
|
|||
|
|
|
|||
|
|
class directionChanges():
|
|||
|
|
def __init__(self, filePath):
|
|||
|
|
self.filePath = filePath
|
|||
|
|
|
|||
|
|
# 图片读取
|
|||
|
|
def imageRead(self):
|
|||
|
|
face3 = Image.open(self.filePath)
|
|||
|
|
return face3
|
|||
|
|
|
|||
|
|
# 图片上矩阵化
|
|||
|
|
def imageMatrix(self):
|
|||
|
|
im3 = np.array(self.imageRead())
|
|||
|
|
return im3
|
|||
|
|
|
|||
|
|
def imageRepetition(self):
|
|||
|
|
# 图像重复
|
|||
|
|
# 先水平方向两次，再垂直方向3次
|
|||
|
|
plt1 = np.concatenate((self.imageMatrix(), self.imageMatrix()), axis=1)
|
|||
|
|
plt2 = np.concatenate((plt1, plt1, plt1), axis=0)
|
|||
|
|
plt.imshow(plt2)
|
|||
|
|
plt.show()
|
|||
|
|
# # 先垂直方向转3次,再水平方向转4次
|
|||
|
|
plt3 = np.concatenate((self.imageMatrix(), self.imageMatrix(), self.imageMatrix()), axis=0)
|
|||
|
|
plt4 = np.concatenate((plt3, plt3, plt3, plt3), axis=1)
|
|||
|
|
plt.imshow(plt4)
|
|||
|
|
plt.show()
|
|||
|
|
return plt2
|
|||
|
|
|
|||
|
|
# 图像水平镜面对称
|
|||
|
|
def mirrorSymmetry(self):
|
|||
|
|
plt6 = self.imageMatrix()[::-1]
|
|||
|
|
plt.imshow(plt6)
|
|||
|
|
plt.show()
|
|||
|
|
return plt6
|
|||
|
|
|
|||
|
|
# 垂直对称(右旋转)
|
|||
|
|
def verticalSummetry_right(self):
|
|||
|
|
plt5 = self.imageMatrix().swapaxes(1, 0)
|
|||
|
|
plt.imshow(plt5[:, ::-1])
|
|||
|
|
plt.show()
|
|||
|
|
return plt5
|
|||
|
|
|
|||
|
|
# 垂直镜面(左旋转)
|
|||
|
|
# transpose函数的作用就是调换数组的行列值的索引值，类似于求矩阵的转置
|
|||
|
|
def verticalSummetry_lift(self):
|
|||
|
|
plt7 = self.imageMatrix().transpose(1, 0, 2) # x,y轴转置
|
|||
|
|
plt.imshow(plt7[::-1])
|
|||
|
|
plt.show()
|
|||
|
|
return plt7
|
|||
|
|
|
|||
|
|
# 保存图片
|
|||
|
|
def img_Save(self):
|
|||
|
|
root = "E:\\桌面\\Python_Picture_Analysis\\data\\"
|
|||
|
|
if not os.path.exists(root):
|
|||
|
|
os.mkdir(root)
|
|||
|
|
else:
|
|||
|
|
Image.fromarray(np.uint8(self.imageRepetition())).save(root+'picture4'+'.jpg')
|
|||
|
|
Image.fromarray(np.uint8(self.mirrorSymmetry())).save(root+'picture5'+'.jpg')
|
|||
|
|
Image.fromarray(np.uint8(self.verticalSummetry_right())).save(root+'picture6'+'.jpg')
|
|||
|
|
Image.fromarray(np.uint8(self.verticalSummetry_lift())).save(root+'picture7'+'.jpg')
|
|||
|
|
|
|||
|
|
|