65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# @Time : 2022-7-24 0024 16:42
|
|
# @Author : Qing
|
|
# @Email : derighoid@gmail.com
|
|
# @File : Modify_Pixels.py
|
|
# @Software: PyCharm
|
|
|
|
import os
|
|
|
|
# 导入库
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
|
|
class modfifyImage():
|
|
def __init__(self, filePath):
|
|
self.filePath = filePath
|
|
|
|
# 图片读取
|
|
def imageRead(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.imageRead())
|
|
return im
|
|
|
|
# 图片像素反转
|
|
def pixInversion(self):
|
|
im2 = 255 - self.imageMatrix()
|
|
plt.figure()
|
|
plt.imshow(im2)
|
|
plt.show()
|
|
return im2
|
|
|
|
# 显示图像三通道颜色
|
|
def img_RGB(self):
|
|
root="E:\\桌面\\Python_Picture_Analysis\\data\\" # 图片保存地址
|
|
red = self.imageMatrix().copy()
|
|
red[:, :, 1:3] = 0
|
|
green = self.imageMatrix().copy()
|
|
green[:, :, ::2] = 0
|
|
blue = self.imageMatrix().copy()
|
|
blue[:, :, :2] = 0
|
|
x, y = plt.subplots(2, 2)
|
|
x.set_size_inches(10, 10)
|
|
y[0, 0].imshow(self.imageMatrix())
|
|
y[0, 1].imshow(red)
|
|
y[1, 0].imshow(green)
|
|
y[1, 1].imshow(blue)
|
|
# 图片保存
|
|
Image.fromarray(np.uint8(red)).save(root + 'picture1' + '.jpg')
|
|
Image.fromarray(np.uint8(blue)).save(root + 'picture2' + '.jpg')
|
|
Image.fromarray(np.uint8(green)).save(root + 'picture3' + '.jpg')
|
|
plt.show()
|
|
return red,green,blue
|
|
|
|
# 获得该点的类型
|
|
def get_imgType(self, x, y):
|
|
return type(self.imageMatrix()[x, y]) |