commit intermediaire

cache.py

@@ -6,12 +6,19 @@
 |=================================================|
 """
 
+import cv2
+
 def cache_img(support, img_a_cacher):
     """
+    entree
+    ------
     :param support: L'image que l'on voit a la fin.
     :param img_a_cacher: L'image que l'on veut cacher.
     :type support: np.ndarray
     :type img_a_cacher: np.ndarray
+    
+	sortie
+	------
     :return: L'image contenant le support et l'image a cacher.
     :rtype: np.ndarray
     """
@@ -20,5 +27,4 @@ def cache_img(support, img_a_cacher):
     # Retirer les bits de poids faible de l'image à cacher et les mettre au format 4 bits, ie rendre les poids forts en faible
     # Remplacer les 4 bits de poids faible du support par les 4 bits de poids fort de l'image à cacher
     # Retourner l'image ainsi obtenue
-
-    return (support&0b11110000) + (img_a_cacher>>4)
+    return (support & 0b11110000) + (cv2.resize(img_a_cacher, suport.shape[:2], interpolation=cv2.INTER_AREA) >> 4)

main.py

@@ -27,7 +27,7 @@ def choisir(titre):
     if not impath:
         sys.stderr.write("Il faut selectionner un fichier!\n")
         return
-    image = cv2.imread(impath)
+    image = cv2.imread(impath, cv2.IMREAD_COLOR)
     if isinstance(image, type(None)):
         sys.stderr.write("L'image est corrompue!\n")
         return
@@ -41,8 +41,8 @@ def main():
     devoile = trouve.trouve_img(mixe)
 
     repertoire = tkinter.filedialog.askdirectory(title="Repertoire d'enregistrement")
-    cv2.imwrite(os.path.join(repertoire, "image_mixe.png"), mixe)
-    cv2.imwrite(os.path.join(repertoire, "image_disimulee.png"), devoile)
+    cv2.imwrite(os.path.join(repertoire, "lemoine_mixe.png"), mixe)
+    cv2.imwrite(os.path.join(repertoire, "lemoine_dissimulee.png"), devoile)
     
 
 if __name__ == "__main__":