Spaces:
Running
Running
Create stegan2.py
Browse files- stegan2.py +106 -0
stegan2.py
ADDED
@@ -0,0 +1,106 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import base64
|
2 |
+
import cv2
|
3 |
+
import numpy as np
|
4 |
+
#import pandas as pd
|
5 |
+
from PIL import Image
|
6 |
+
import os
|
7 |
+
import uuid as uniq
|
8 |
+
from qr import make_qr
|
9 |
+
import math
|
10 |
+
|
11 |
+
|
12 |
+
def to_bin(data):
|
13 |
+
"""Convert `data` to binary format as string"""
|
14 |
+
if isinstance(data, str):
|
15 |
+
return ''.join([ format(ord(i), "08b") for i in data ])
|
16 |
+
elif isinstance(data, bytes):
|
17 |
+
return ''.join([ format(i, "08b") for i in data ])
|
18 |
+
elif isinstance(data, np.ndarray):
|
19 |
+
return [ format(i, "08b") for i in data ]
|
20 |
+
elif isinstance(data, int) or isinstance(data, np.uint8):
|
21 |
+
return format(data, "08b")
|
22 |
+
else:
|
23 |
+
raise TypeError("Type not supported.")
|
24 |
+
def decode(image_name,txt=None):
|
25 |
+
BGRimage = cv2.imread(image_name)
|
26 |
+
image = cv2.cvtColor(BGRimage, cv2.COLOR_BGR2RGB)
|
27 |
+
binary_data = ""
|
28 |
+
for row in image:
|
29 |
+
for pixel in row:
|
30 |
+
r, g, b = to_bin(pixel)
|
31 |
+
binary_data += r[-1]
|
32 |
+
binary_data += g[-1]
|
33 |
+
binary_data += b[-1]
|
34 |
+
all_bytes = [ binary_data[i: i+8] for i in range(0, len(binary_data), 8) ]
|
35 |
+
decoded_data = ""
|
36 |
+
for byte in all_bytes:
|
37 |
+
decoded_data += chr(int(byte, 2))
|
38 |
+
if decoded_data[-5:] == "=====":
|
39 |
+
break
|
40 |
+
this = decoded_data[:-5].split("#####",1)[0]
|
41 |
+
this = eval(this)
|
42 |
+
enc_in=this
|
43 |
+
return this
|
44 |
+
|
45 |
+
def encode(image_name, secret_data,txt=None):
|
46 |
+
BGRimage = cv2.imread(image_name)
|
47 |
+
image = cv2.cvtColor(BGRimage, cv2.COLOR_BGR2RGB)
|
48 |
+
n_bytes = image.shape[0] * image.shape[1] * 3 // 8
|
49 |
+
print("[*] Maximum bytes to encode:", n_bytes)
|
50 |
+
secret_data1=secret_data
|
51 |
+
#secret_data1=f'{secret_data}#{resultp}'
|
52 |
+
|
53 |
+
while True:
|
54 |
+
if len(secret_data1)+5 < (n_bytes):
|
55 |
+
secret_data1 = f'{secret_data1}#####'
|
56 |
+
elif len(secret_data1)+5 >= (n_bytes):
|
57 |
+
break
|
58 |
+
secret_data = secret_data1
|
59 |
+
if len(secret_data) > n_bytes:
|
60 |
+
return image_name, gr.Markdown.update("""<center><h3>Input image is too large""")
|
61 |
+
secret_data += "====="
|
62 |
+
data_index = 0
|
63 |
+
binary_secret_data = to_bin(secret_data)
|
64 |
+
data_len = len(binary_secret_data)
|
65 |
+
for row in image:
|
66 |
+
for pixel in row:
|
67 |
+
r, g, b = to_bin(pixel)
|
68 |
+
if data_index < data_len:
|
69 |
+
pixel[0] = int(r[:-1] + binary_secret_data[data_index], 2)
|
70 |
+
data_index += 1
|
71 |
+
if data_index < data_len:
|
72 |
+
pixel[1] = int(g[:-1] + binary_secret_data[data_index], 2)
|
73 |
+
data_index += 1
|
74 |
+
if data_index < data_len:
|
75 |
+
pixel[2] = int(b[:-1] + binary_secret_data[data_index], 2)
|
76 |
+
data_index += 1
|
77 |
+
if data_index >= data_len:
|
78 |
+
break
|
79 |
+
return image
|
80 |
+
def conv_im(qr_link,data):
|
81 |
+
uniqnum = uniq.uuid4()
|
82 |
+
|
83 |
+
byte_size = len(data)
|
84 |
+
print (f'bytes:{byte_size}')
|
85 |
+
data_pixels = byte_size*4
|
86 |
+
print (f'pixels:{data_pixels}')
|
87 |
+
#data_sq=data_pixels/2
|
88 |
+
data_sq = int(math.sqrt(data_pixels))
|
89 |
+
data_pad = data_sq+100
|
90 |
+
print (f'square image:{data_pad}x{data_pad}')
|
91 |
+
|
92 |
+
qr_im = make_qr(txt=qr_link)
|
93 |
+
img1 = Image.open(qr_im)
|
94 |
+
imgw = img1.size[0]
|
95 |
+
imgh = img1.size[1]
|
96 |
+
print (f'qr Size:{img1.size}')
|
97 |
+
#img1.thumbnail((imgw*4,imgh*4), Image.Resampling.LANCZOS)
|
98 |
+
img1 = img1.resize((int(data_pad),int(data_pad)), Image.Resampling.LANCZOS)
|
99 |
+
print (img1.size)
|
100 |
+
img1.save(f'tmpim{uniqnum}.png')
|
101 |
+
|
102 |
+
with open(f'tmpim{uniqnum}.png', "rb") as image_file:
|
103 |
+
encoded_string = base64.b64encode(image_file.read())
|
104 |
+
image_file.close()
|
105 |
+
im_out = encode(f'tmpim{uniqnum}.png',data)
|
106 |
+
return im_out
|