Шифрование RGB изображений

"""
Функция для генерации ключа с использованием хаотической карты для процесса шифрования
"""
 
def keygen(x,r,size):
 
    key=[]
 
    for i in range(size):
 
        x=r*x*(1-x) #logistic map
        key.append(int((x*pow(10, 16))%256)) #key=(x*10^16)%256
 
    return key
 
# print(keygen(0.01, 3.915, 10))
 
 
""""
Шифрование изображения
"""
import keygen as kg
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
 
def enc():
    # Чтение изображения
    img = mpimg.imread('C:\\Users\\acer\\Desktop\\MyProject\\images\\2.bmp')
    # Generating chaotic keys
    height = img.shape[0]
    width = img.shape[1]
    key=kg.keygen(0.01, 3.915, height*width)
    #print(key)
    #Шифрование-Замена с помощью XOR
    z=0
   #Изображения RGB обычно хранятся в виде 3-мерных массивов 8-битных целых чисел без знака
    enimg = np.zeros(shape=[height, width, 3], dtype=np.uint8)
    for i in range(height):
        for j in range(width):
            enimg[i,j]=img[i,j]^key[z]
            z+=1
    plt.imshow(enimg)
    plt.show()
    #plt.imsave('C:\\Users\\acer\\Desktop\\MyProject\\images\\2enc.bmp', enimg)
    #return enimg

 
""""
Шифрование изображения
"""
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
import cv2
 
#Функция для генерации ключа с использованием хаотической карты для процесса шифрования
def keygen(x,r,size):
 
    key=[]
 
    for i in range(size):
 
        x=r*x*(1-x) #logistic map
        key.append(int((x*pow(10, 16))%256)) #key=(x*10^16)%256
 
    return key
# print(keygen(0.01, 3.915, 10))
 
 
    # Чтение изображения
img = mpimg.imread('C:\\Users\\77782\\Desktop\\MyProject\\images\\2.bmp')
# Generating chaotic keys
height = img.shape[0]
width = img.shape[1]
key=keygen(0.01, 3.915, height*width)
print(key)
#Шифрование-Замена с помощью XOR
z=0
#Изображения RGB обычно хранятся в виде 3-мерных массивов 8-битных целых чисел без знака
 
blue, green, red = cv2.split(img)
print("Image splited")
enBimg = np.zeros(shape=[height, width, 3], dtype=np.uint8)
enGimg = np.zeros(shape=[height, width, 3], dtype=np.uint8)
enRimg = np.zeros(shape=[height, width, 3], dtype=np.uint8)
enimg = np.zeros(shape=[height, width, 3], dtype=np.uint8)
for i in range(height):
    for j in range(width):
        enBimg[i,j]=blue[i,j]^key[z]
        enGimg[i,j]=green[i,j]^key[z]
        enRimg[i,j]=red[i,j]^key[z]
        z+=1
print("image merged")
enimg= cv2.merge((enBimg, enGimg, enRimg))
 
h,w = enimg.shape[:2]
print(enimg.shape)
plt.imshow(enimg,cmap='gray')
plt.show()
 
#plt.imshow(enRimg)
#plt.show()
 
plt.imsave('C:\\Users\\77782\\Desktop\\MyProject\\images\\Renc.bmp',enRimg)
plt.imsave('C:\\Users\\77782\\Desktop\\MyProject\\images\\Benc.bmp', enBimg)
plt.imsave('C:\\Users\\77782\\Desktop\\MyProject\\images\\Genc.bmp', enGimg)
#cv2.imwrite('C:\\Users\\77782\\Desktop\\MyProject\\images\\1MA.bmp',enimg)
#return enimg