replace PIL with cv2

Author: Nibba2018

Python/Image-Steganography/hide.py

@@ -1,4 +1,4 @@
-from PIL import Image
+import cv2
 import numpy as np
 import sys
 from cryptography.fernet import Fernet
@@ -9,6 +9,17 @@
     sys.exit()
 
 
+def message_to_binary(message):
+    if type(message) == str:
+        return ''.join([ format(ord(i), "08b") for i in message ])
+    elif type(message) == bytes or type(message) == np.ndarray:
+        return [ format(i, "08b") for i in message ]
+    elif type(message) == int or type(message) == np.uint8:
+        return format(message, "08b")
+    else:
+        raise TypeError("Input type not supported")
+
+
 def encrypt_message(message):
     key = Fernet.generate_key()
     encrypted_message = Fernet(key).encrypt(message.encode())
@@ -19,23 +30,23 @@ def encrypt_message(message):
 def hide_info(image, message):
 
     enc_message, key = encrypt_message(message)
-    image = np.array(image)
+    enc_message = enc_message.decode()
 
     max_bytes = image.shape[0] * image.shape[1] * 3//8
 
     if len(enc_message) > max_bytes:
         raise ValueError("Insufficient bytes, provide bigger image or shorter message.")
 
-    enc_message += "#####".encode()
+    enc_message += "#####"
 
     data_index = 0
-    bin_enc_message = [ format(i, "08b") for i in enc_message]
+    bin_enc_message = message_to_binary(enc_message)
 
     data_len = len(bin_enc_message)
 
     for values in image:
         for pixel in values:
-            r, g, b = [format(i, "08b") for i in pixel]
+            r, g, b = message_to_binary(pixel)
 
             if data_index < data_len:
                 pixel[0] = int(r[:-1] + bin_enc_message[data_index], 2)
@@ -55,17 +66,17 @@ def hide_info(image, message):
     with open("key", "wb") as f:
         f.write(key)
 
-    return Image.fromarray(image)
+    return image
 
 
 image_path = sys.argv[1]
 message_path = sys.argv[2]
 
-source_image = Image.open(image_path, "r")
+source_image = cv2.imread(image_path)
 with open(message_path, "r") as f:
     message = f.read()
 
 secret_image = hide_info(source_image, message)
 secret_image_path = image_path.split(".")
 secret_image_path = secret_image_path[-2]+"_secret."+secret_image_path[-1]
-secret_image.save(secret_image_path)
+cv2.imwrite(secret_image_path, secret_image)

Python/Image-Steganography/key

@@ -1 +1 @@
-VVagxe87pFTfpWsr78zh5s6cyyzXWD4_-YMpJpB-vvI=
+UJRsXUb0QISvEQwbyl4M0FVhl-Xt1X_mWyjQClbT3KE=

Python/Image-Steganography/retrieve.py

@@ -1,5 +1,5 @@
 from cryptography.fernet import Fernet
-from PIL import Image
+import cv2
 import numpy as np
 import sys
 
@@ -8,8 +8,15 @@
     sys.exit()
 
 
-image_path = sys.argv[1]
-key_path = sys.argv[2]
+def message_to_binary(message):
+    if type(message) == str:
+        return ''.join([ format(ord(i), "08b") for i in message ])
+    elif type(message) == bytes or type(message) == np.ndarray:
+        return [ format(i, "08b") for i in message ]
+    elif type(message) == int or type(message) == np.uint8:
+        return format(message, "08b")
+    else:
+        raise TypeError("Input type not supported")
 
 
 def decrypt_message(enc_message, key):
@@ -20,11 +27,10 @@ def decrypt_message(enc_message, key):
 def retrieve_info(image, key):
 
     binary_data = ""
-    image = np.array(image)
 
     for values in image:
         for pixel in values:
-            r, g, b = [format(i, "08b") for i in pixel]
+            r, g, b = message_to_binary(pixel)
             binary_data += r[-1]
             binary_data += g[-1]
             binary_data += b[-1]
@@ -34,15 +40,17 @@ def retrieve_info(image, key):
     decoded_data = ""
     for byte in all_bytes:
         decoded_data += chr(int(byte, 2))
-        if decoded_data[-5:] == "#####".encode():
+        if decoded_data[-5:] == "#####":
             break
 
     message = decrypt_message(decoded_data[:-5].encode(), key)
 
-    return decoded_data[:-5]
+    return message.decode()
 
+image_path = sys.argv[1]
+key_path = sys.argv[2]
 
-steg_image = Image.open(image_path, 'r')
+steg_image = cv2.imread(image_path)
 with open(key_path, "rb") as f:
     key = f.read()