September 11, 2024

Python Tutorials

Image Steganography Using Python

Image steganography involves hiding secret information within an image file. Using Python, you can implement steganography to embed text or data within an image in a way that’s imperceptible to the human eye. Below is a step-by-step guide to performing image steganography using Python.

1. Install Required Libraries

You will need the Pillow library for image processing. Install it using pip:

pip install pillow

2. Create Functions for Steganography

Here’s how to create functions to embed and extract text from an image.

2.1. Function to Embed Text in an Image

from PIL import Image

def encode_image(image_path, secret_text, output_path):
    img = Image.open(image_path)
    encoded_img = img.copy()
    width, height = img.size
    
    # Append the length of the secret text to the end of the text
    secret_text += chr(len(secret_text))
    
    # Convert the secret text into binary
    binary_text = ''.join(format(ord(char), '08b') for char in secret_text)
    
    # Ensure the text fits in the image
    if len(binary_text) > width * height * 3:
        raise ValueError("The secret text is too long to encode in this image.")
    
    binary_index = 0
    pixels = encoded_img.load()
    
    for x in range(width):
        for y in range(height):
            pixel = list(pixels[x, y])
            for i in range(3):
                if binary_index < len(binary_text):
                    pixel[i] = int(format(pixel[i], '08b')[:-1] + binary_text[binary_index], 2)
                    binary_index += 1
            pixels[x, y] = tuple(pixel)
    
    encoded_img.save(output_path)

print("Image encoding complete.")

2.2. Function to Extract Text from an Image

def decode_image(image_path):
    img = Image.open(image_path)
    width, height = img.size
    binary_text = ""
    
    pixels = img.load()
    
    for x in range(width):
        for y in range(height):
            pixel = pixels[x, y]
            for i in range(3):
                binary_text += format(pixel[i], '08b')[-1]
                
    # Split the binary text into characters and decode
    char_bits = [binary_text[i:i+8] for i in range(0, len(binary_text), 8)]
    decoded_chars = [chr(int(bits, 2)) for bits in char_bits]
    
    # Find where the length byte is stored
    length = ord(decoded_chars[-1])
    secret_text = ''.join(decoded_chars[:-1])
    
    return secret_text[:length]

print("Image decoding complete.")

3. Example Usage

Here’s how to use the above functions to encode and decode text within an image:

# Encode a message into an image
encode_image('input_image.png', 'Secret Message', 'encoded_image.png')

# Decode the message from the image
message = decode_image('encoded_image.png')
print(f'Decoded message: {message}')

4. Considerations and Improvements

  • Image Format: The example provided uses PNG images, as they are lossless and preserve the image data. Avoid using formats like JPEG which may alter the image.
  • Data Size: Ensure the amount of data you want to encode fits within the image. Larger images can hold more data.
  • Error Handling: Add error handling to manage cases where text cannot be embedded or decoded properly.
  • Encryption: For added security, consider encrypting the secret text before embedding it in the image.

This guide provides a basic approach to image steganography in Python. You can enhance it further based on your requirements and ensure robust security measures are in place for sensitive information.

You May Like