Blame view

test_steg.py 1.13 KB
548d95dc   Chunk   steganography(F3 ...
1
2
3
4
5
6
7
__author__ = 'chunk'

import numpy as np
import pylab as P
import pylab as plt

import mjpeg
8cfc1a23   Chunk   F5 half-finished.
8
from  msteg.steganography import LSB, F3, F4, F5
548d95dc   Chunk   steganography(F3 ...
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from common import *


timer = Timer()

sample = [[7, 12, 14, -12, 1, 0, -1, 0],
          [6, 5, -10, 0, 6, 0, 0, 0],
          [0, 6, -5, 4, 0, -1, 0, 0],
          [0, -3, 0, 1, -1, 0, 0, 0],
          [-3, 5, 0, 0, 0, 0, 0, 0],
          [2, -1, 0, 0, 0, 0, 0, 0],
          [0, 0, 0, 0, 0, 0, 0, 0],
          [0, 0, 0, 0, 0, 0, 0, 0]]

8cfc1a23   Chunk   F5 half-finished.
23
24
sample_key = [46812L, 20559L, 31360L, 16681L, 27536L, 39553L, 5427L, 63029L, 56572L, 36476L, 25695L, 61908L, 63014L,
              5908L, 59816L, 56765L]
6cbb3879   Chunk   F4 updated.
25

548d95dc   Chunk   steganography(F3 ...
26
27
28
txtsample = [116, 104, 105, 115, 32, 105, 115, 32, 116, 111, 32, 98, 101, 32, 101, 109, 98, 101, 100, 101, 100, 46, 10]

if __name__ == '__main__':
dceec280   Chunk   get capacity.
29
    # f3test = F4.F4(sample_key)
8cfc1a23   Chunk   F5 half-finished.
30
    f3test = F5.F5(sample_key, 1)
04cd1acf   Chunk   before 开题答辩~~~
31
32
33
    # f3test.embed_raw_data("res/test3.jpg", "res/embeded", "res/steged.jpg")
    f3test.embed_raw_data("res/thulib2.jpg", "res/lena64gray.jpg", "res/steged.jpg")

8cfc1a23   Chunk   F5 half-finished.
34
35
    # f3test2 = F4.F4(sample_key)
    f3test.extract_raw_data("res/steged.jpg", "res/extracted.jpg")
04cd1acf   Chunk   before 开题答辩~~~
36
    print f3test.get_key()
033d3b0d   Chunk   staged.
37
    pass
dceec280   Chunk   get capacity.

548d95dc   Chunk   steganography(F3 ...