Python unpack_img_to_bits Examples

Example #1

File: retrieve.py Project: qll/shit

def decode_msg_from_bit(img_array, stegobit):
    img_bit_array = util.unpack_img_to_bits(img_array)

    bitmsg = [byte[stegobit] for byte in img_bit_array]

    msg = np.packbits(bitmsg)
    return''.join(chr(x) for x in msg if x != 0)

Example #2

File: retrieve.py Project: toanou/shit

def decode_msg_from_bit(img_array, stegobit):
    img_bit_array = util.unpack_img_to_bits(img_array)

    bitmsg = [byte[stegobit] for byte in img_bit_array]

    msg = np.packbits(bitmsg)
    return ''.join(chr(x) for x in msg if x != 0)

Example #3

File: hide.py Project: qll/shit

def encode_msg_in_bit(img_array, msg, stegobit):
    assert len(msg) * 8 < img_array.size, 'The message is too long for this image.'

    binmsg = np.array(convert_msg_to_bin(msg), dtype= np.uint8)
    zeros = np.zeros(img_array.size - binmsg.size, dtype=np.uint8)
    padmsg = np.append(binmsg, zeros)

    img_bit_array = util.unpack_img_to_bits(img_array.astype(np.uint8))

    #manipulation
    for byte, msg_bit in zip(img_bit_array, padmsg): 
        byte[stegobit] = msg_bit

    return util.pack_bits_to_img(img_bit_array, img_array)

Example #4

def encode_msg_in_bit(img_array, msg, stegobit):
    assert len(
        msg) * 8 < img_array.size, 'The message is too long for this image.'

    binmsg = np.array(convert_msg_to_bin(msg), dtype=np.uint8)
    zeros = np.zeros(img_array.size - binmsg.size, dtype=np.uint8)
    padmsg = np.append(binmsg, zeros)

    img_bit_array = util.unpack_img_to_bits(img_array.astype(np.uint8))

    #manipulation
    for byte, msg_bit in zip(img_bit_array, padmsg):
        byte[stegobit] = msg_bit

    return util.pack_bits_to_img(img_bit_array, img_array)