DIP_Watermark_Evasion_TMLR/count_decode_results.py at main · sun-umn/DIP_Watermark_Evasion_TMLR · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import numpy as np
import math, os
import pickle


if __name__ == "__main__":

    # Count File numbers
    watermarker = "StegaStamp"
    dataset = "DiffusionDB"

    evader = "corrupters"
    archs = ["brightness", "contrast", "gaussian_noise", "jpeg", "bm3d"]
    # archs = ["brightness", "contrast", "gaussian_noise", "jpeg"]
    for arch in archs:
        result_dir = os.path.join(
            "Result-Decoded", watermarker, dataset, evader, arch
        )
        file_list = [f for f in os.listdir(result_dir)]
        print("{} - {} - {} - {}".format(watermarker, dataset, evader, arch))
        print("Number of files processed: ", len(file_list))
        print()

    evaders = ["diffuser", "diffpure"]
    arch = "dummy"
    for evader in evaders:
        result_dir = os.path.join(
            "Result-Decoded", watermarker, dataset, evader, arch
        )
        file_list = [f for f in os.listdir(result_dir)]
        print("{} - {} - {} - {}".format(watermarker, dataset, evader, arch))
        print("Number of files processed: ", len(file_list))
        print()

    evader = "vae"
    arch = "cheng2020-anchor"
    result_dir = os.path.join(
        "Result-Decoded", watermarker, dataset, evader, arch
    )
    file_list = [f for f in os.listdir(result_dir)]
    print("{} - {} - {} - {}".format(watermarker, dataset, evader, arch))
    print("Number of files processed: ", len(file_list))
    print()

    evader = "dip"
    arch = "vanila"
    result_dir = os.path.join(
        "Result-Decoded", watermarker, dataset, evader, arch
    )
    file_list = [f for f in os.listdir(result_dir)]
    print("{} - {} - {} - {}".format(watermarker, dataset, evader, arch))
    print("Number of files processed: ", len(file_list))
    print()
    print("Completed.")