| | import cv2 |
| | import numpy as np |
| | from PIL import Image, PngImagePlugin |
| |
|
| | def png_encode(im_name,extra): |
| | |
| |
|
| | im = Image.open(im_name) |
| |
|
| | info = PngImagePlugin.PngInfo() |
| | info.add_text("TXT", extra) |
| | im.save("test.png",pnginfo=info) |
| | test = Image.open("test.png") |
| | print(test.text) |
| | return("test.png","") |
| |
|
| |
|
| | def to_bin(data): |
| | """Convert `data` to binary format as string""" |
| | if isinstance(data, str): |
| | return ''.join([ format(ord(i), "08b") for i in data ]) |
| | elif isinstance(data, bytes): |
| | return ''.join([ format(i, "08b") for i in data ]) |
| | elif isinstance(data, np.ndarray): |
| | return [ format(i, "08b") for i in data ] |
| | elif isinstance(data, int) or isinstance(data, np.uint8): |
| | return format(data, "08b") |
| | else: |
| | raise TypeError("Type not supported.") |
| | def decode(image_name,txt=None): |
| | try: |
| | BGRimage = cv2.imread(f"{image_name}") |
| | image = cv2.cvtColor(BGRimage, cv2.COLOR_BGR2RGB) |
| | binary_data = "" |
| | for row in image: |
| | for pixel in row: |
| | r, g, b = to_bin(pixel) |
| | binary_data += r[-1] |
| | binary_data += g[-1] |
| | binary_data += b[-1] |
| | all_bytes = [ binary_data[i: i+8] for i in range(0, len(binary_data), 8) ] |
| | decoded_data = "" |
| | for byte in all_bytes: |
| | decoded_data += chr(int(byte, 2)) |
| | if decoded_data[-5:] == "=====": |
| | break |
| | this = decoded_data[:-5].split("#####",1)[0] |
| | |
| | except Exception as e: |
| | this = e |
| | return this |
| |
|
| | def encode(image_name, secret_data,txt=None): |
| | msg="" |
| | BGRimage = cv2.imread(f"{image_name}") |
| | image = cv2.cvtColor(BGRimage, cv2.COLOR_BGR2RGB) |
| | n_bytes = image.shape[0] * image.shape[1] * 3 // 8 |
| | print("[*] Maximum bytes to encode:", n_bytes) |
| | secret_data1=secret_data |
| | while True: |
| | if len(secret_data1)+5 < (n_bytes): |
| | secret_data1 = f'{secret_data1}#####' |
| | elif len(secret_data1)+5 >= (n_bytes): |
| | break |
| | secret_data = secret_data1 |
| | if len(secret_data) > n_bytes: |
| | msg ="Watermark is too large for Image Size" |
| | return image_name,msg |
| | secret_data += "=====" |
| | data_index = 0 |
| | binary_secret_data = to_bin(secret_data) |
| | data_len = len(binary_secret_data) |
| | for row in image: |
| | for pixel in row: |
| | r, g, b = to_bin(pixel) |
| | if data_index < data_len: |
| | pixel[0] = int(r[:-1] + binary_secret_data[data_index], 2) |
| | data_index += 1 |
| | if data_index < data_len: |
| | pixel[1] = int(g[:-1] + binary_secret_data[data_index], 2) |
| | data_index += 1 |
| | if data_index < data_len: |
| | pixel[2] = int(b[:-1] + binary_secret_data[data_index], 2) |
| | data_index += 1 |
| | if data_index >= data_len: |
| | break |
| | return image,msg |
