74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# @Time : 2022-7-24 0024 16:00
|
|
# @Author : Qing
|
|
# @Email : derighoid@gmail.com
|
|
# @File : Picture_Reading.py
|
|
# @Software: PyCharm
|
|
|
|
# 导入库
|
|
import os.path
|
|
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
|
|
# 建立图片读取类
|
|
class pictureRead(object):
|
|
def __init__(self, filePath):
|
|
self.filePath = filePath
|
|
|
|
# 图片读取
|
|
def fileRead(self):
|
|
if self.filePath == " ":
|
|
print("Please full in the file path")
|
|
else:
|
|
face = Image.open(self.filePath)
|
|
return face
|
|
|
|
# 图片转矩阵
|
|
def imageMatrix(self):
|
|
im = np.array(self.fileRead())
|
|
return im
|
|
|
|
# 图片转灰色
|
|
def imageGrey(self):
|
|
im_gray = np.array(self.fileRead().convert("L"))
|
|
return im_gray
|
|
|
|
# 打印图片类型
|
|
def get_imageType(self):
|
|
print(type(self.imageGrey()))
|
|
|
|
# 打印图片矩阵元素类型
|
|
def get_imageDtype(self):
|
|
print(self.imageGrey().dtype)
|
|
|
|
# 打印图片大小
|
|
def get_imageSize(self):
|
|
print(self.imageMatrix().shape)
|
|
|
|
# 图片plt展示
|
|
def plt_image(self):
|
|
# 原图片显示
|
|
plt.imshow(self.imageMatrix())
|
|
plt.show()
|
|
# 灰色图片显示
|
|
plt.imshow(self.imageGrey())
|
|
plt.show()
|
|
|
|
# 图片保存
|
|
def image_save(self):
|
|
path = 'E:\\桌面\\Python_Picture_Analysis\\data\\numpy_red.png'
|
|
root="E:\\桌面\\Python_Picture_Analysis\\data\\"
|
|
if not os.path.exists(root):
|
|
os.mkdir(root) # 判断是否存在文件夹
|
|
else:
|
|
if os.path.exists(path): # 判断是否已经存在该文件
|
|
print("File already exists")
|
|
else:
|
|
Image.fromarray(np.uint8(self.imageGrey())).save(path)
|
|
print("Save successfully")
|
|
|
|
|