Blame view

msteg/steganalysis/MPB.py 5.34 KB
26e2fe9f   Chunk   MPB steganalysis ...
1
2
3
4
5
6
7
8
__author__ = 'chunk'
"""
Yun Q. Shi, et al - A Markov Process Based Approach to Effective Attacking JPEG Steganography
"""

import time
import math
import numpy as np
04cd1acf   Chunk   before 开题答辩~~~
9
from msteg.StegBase import *
09268ae3   Chunk   An Algorithm-Bug ...
10
11
12
13
14
import mjsteg
import jpegObj
from common import *

import csv
26e2fe9f   Chunk   MPB steganalysis ...
15
16
17
18
import json
import pickle
from sklearn import svm

6d219855   Chunk   MPB finished. The...
19
base_dir = '/home/hadoop/data/HeadShoulder/'
26e2fe9f   Chunk   MPB steganalysis ...
20
21

class MPB(StegBase):
f4b5291c   Chunk   Qaulity Calculati...
22
23
24
25
    """
    Markov Process Based Steganalyasis Algo.
    """

26e2fe9f   Chunk   MPB steganalysis ...
26
27
    def __init__(self):
        StegBase.__init__(self, sample_key)
6d219855   Chunk   MPB finished. The...
28

26e2fe9f   Chunk   MPB steganalysis ...
29
30
31
32
33
34
35
    def get_trans_prob_mat_orig(self, ciq, T=4):
        """
        Original!
        Calculate Transition Probability Matrix.

        :param ciq: jpeg DCT coeff matrix, 2-D numpy array of int16 (pre-abs)
        :param T: signed integer, usually 1~7
6d219855   Chunk   MPB finished. The...
36
37
        :return: TPM - 3-D tensor, numpy array of size (2*T+1, 2*T+1, 4)
        """
26e2fe9f   Chunk   MPB steganalysis ...
38
        ciq = np.absolute(ciq).clip(0, T)
6d219855   Chunk   MPB finished. The...
39
        TPM = np.zeros((2 * T + 1, 2 * T + 1, 4), np.float64)
26e2fe9f   Chunk   MPB steganalysis ...
40
41
42
43
44
45
46
47
        # Fh = np.diff(ciq, axis=-1)
        # Fv = np.diff(ciq, axis=0)
        Fh = ciq[:-1, :-1] - ciq[:-1, 1:]
        Fv = ciq[:-1, :-1] - ciq[1:, :-1]
        Fd = ciq[:-1, :-1] - ciq[1:, 1:]
        Fm = ciq[:-1, 1:] - ciq[1:, :-1]

        Fh1 = Fh[:-1, :-1]
09268ae3   Chunk   An Algorithm-Bug ...
48
        Fh2 = Fh[:-1, 1:]
26e2fe9f   Chunk   MPB steganalysis ...
49
50
51

        Fv1 = Fv[:-1, :-1]
        Fv2 = Fv[1:, :-1]
09268ae3   Chunk   An Algorithm-Bug ...
52
53
54
55

        Fd1 = Fd[:-1, :-1]
        Fd2 = Fd[1:, 1:]

26e2fe9f   Chunk   MPB steganalysis ...
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
        Fm1 = Fm[:-1, 1:]
        Fm2 = Fm[1:, :-1]

        # original:(very slow!)
        for n in range(-T, T + 1):
            for m in range(-T, T + 1):
                dh = np.sum(Fh1 == m) * 1.0
                dv = np.sum(Fv1 == m) * 1.0
                dd = np.sum(Fd1 == m) * 1.0
                dm = np.sum(Fm1 == m) * 1.0

                if dh != 0:
                    TPM[m, n, 0] = np.sum(np.logical_and(Fh1 == m, Fh2 == n)) / dh

                if dv != 0:
                    TPM[m, n, 1] = np.sum(np.logical_and(Fv1 == m, Fv2 == n)) / dv

                if dd != 0:
                    TPM[m, n, 2] = np.sum(np.logical_and(Fd1 == m, Fd2 == n)) / dd

                if dm != 0:
                    TPM[m, n, 3] = np.sum(np.logical_and(Fm1 == m, Fm2 == n)) / dm

        # 1.422729s
        return TPM


    def get_trans_prob_mat(self, ciq, T=4):
        """
        Calculate Transition Probability Matrix.

        :param ciq: jpeg DCT coeff matrix, 2-D numpy array of int16 (pre-abs)
        :param T: signed integer, usually 1~7
        :return: TPM - 3-D tensor, numpy array of size (2*T+1, 2*T+1, 4)
        """
        # return self.get_trans_prob_mat_orig(ciq, T)
4a20967b   Chunk   staged.
92
        # timer = Timer()
26e2fe9f   Chunk   MPB steganalysis ...
93
94
95
96
97
98
99
100
        ciq = np.absolute(ciq).clip(0, T)
        TPM = np.zeros((2 * T + 1, 2 * T + 1, 4), np.float64)
        # Fh = np.diff(ciq, axis=-1)
        # Fv = np.diff(ciq, axis=0)
        Fh = ciq[:-1, :-1] - ciq[:-1, 1:]
        Fv = ciq[:-1, :-1] - ciq[1:, :-1]
        Fd = ciq[:-1, :-1] - ciq[1:, 1:]
        Fm = ciq[:-1, 1:] - ciq[1:, :-1]
6d219855   Chunk   MPB finished. The...
101

04cd1acf   Chunk   before 开题答辩~~~
102
        Fh1 = Fh[:-1, :-1]
6d219855   Chunk   MPB finished. The...
103
104
        Fh2 = Fh[:-1, 1:]

26e2fe9f   Chunk   MPB steganalysis ...
105
        Fv1 = Fv[:-1, :-1]
09268ae3   Chunk   An Algorithm-Bug ...
106
107
        Fv2 = Fv[1:, :-1]

26e2fe9f   Chunk   MPB steganalysis ...
108
109
110
        Fd1 = Fd[:-1, :-1]
        Fd2 = Fd[1:, 1:]

09268ae3   Chunk   An Algorithm-Bug ...
111
112
113
114
115
116
117
118
119
120
        Fm1 = Fm[:-1, 1:]
        Fm2 = Fm[1:, :-1]



        # 0.089754s
        # timer.mark()
        # TPM[Fh1.ravel(), Fh2.ravel(), 0] += 1
        # TPM[Fv1.ravel(), Fv2.ravel(), 1] += 1
        # TPM[Fd1.ravel(), Fd2.ravel(), 2] += 1
26e2fe9f   Chunk   MPB steganalysis ...
121
        # TPM[Fm1.ravel(), Fm2.ravel(), 3] += 1
04cd1acf   Chunk   before 开题答辩~~~
122
123
        # timer.report()

26e2fe9f   Chunk   MPB steganalysis ...
124
        # 1.936746s
04cd1acf   Chunk   before 开题答辩~~~
125
126
        # timer.mark()
        for m, n in zip(Fh1.ravel(), Fh2.ravel()):
26e2fe9f   Chunk   MPB steganalysis ...
127
            TPM[m, n, 0] += 1
04cd1acf   Chunk   before 开题答辩~~~
128
129

        for m, n in zip(Fv1.ravel(), Fv2.ravel()):
26e2fe9f   Chunk   MPB steganalysis ...
130
            TPM[m, n, 1] += 1
04cd1acf   Chunk   before 开题答辩~~~
131
132

        for m, n in zip(Fd1.ravel(), Fd2.ravel()):
26e2fe9f   Chunk   MPB steganalysis ...
133
134
135
136
137
138
139
140
141
142
143
            TPM[m, n, 2] += 1

        for m, n in zip(Fm1.ravel(), Fm2.ravel()):
            TPM[m, n, 3] += 1
        # timer.report()

        # 0.057505s
        # timer.mark()
        for m in range(-T, T + 1):
            dh = np.sum(Fh1 == m) * 1.0
            dv = np.sum(Fv1 == m) * 1.0
04cd1acf   Chunk   before 开题答辩~~~
144
145
146
            dd = np.sum(Fd1 == m) * 1.0
            dm = np.sum(Fm1 == m) * 1.0

09268ae3   Chunk   An Algorithm-Bug ...
147
            if dh != 0:
04cd1acf   Chunk   before 开题答辩~~~
148
149
150
151
152
153
154
155
156
                TPM[m, :, 0] /= dh

            if dv != 0:
                TPM[m, :, 1] /= dv

            if dd != 0:
                TPM[m, :, 2] /= dd

            if dm != 0:
26e2fe9f   Chunk   MPB steganalysis ...
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
                TPM[m, :, 3] /= dm
        # timer.report()

        return TPM

    def _load_dataset(self,list_file):
        """
        load jpeg dataset according to a file of file-list.

        :param list_file: a tsv file with each line for a jpeg file path
        :return:(X,Y) for SVM
        """
        X = []
        Y = []
        dict_tagbuf = {}
        dict_dataset = {}

        with open(list_file, 'rb') as tsvfile:
            tsvfile = csv.reader(tsvfile, delimiter='\t')
            for line in tsvfile:
                imgname = line[0] + '.jpg'
                dict_tagbuf[imgname] = line[1]

        dir = base_dir + 'Feat/'
        for path, subdirs, files in os.walk(dir + 'Train/'):
            for name in files:
                featpath = os.path.join(path, name)
                # print featpath
                with open(featpath, 'rb') as featfile:
                    imgname = path.split('/')[-1] + name.replace('.mpb', '.jpg')
                    dict_dataset[imgname] = json.loads(featfile.read())

        for imgname, tag in dict_tagbuf.items():
            tag = 1 if tag == 'True' else 0
            X.append(dict_dataset[imgname])
            Y.append(tag)

6d219855   Chunk   MPB finished. The...
194
        return X, Y
26e2fe9f   Chunk   MPB steganalysis ...

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

6d219855   Chunk   MPB finished. The...

c6c61f81   Chunk   staged.

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

c6c61f81   Chunk   staged.

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...

b69b6985   Chunk   py module refract...

6d219855   Chunk   MPB finished. The...

26e2fe9f   Chunk   MPB steganalysis ...