File size: 880 Bytes
0788e19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from util import mkdir

# directory to store the results
results_dir = './results/'
mkdir(results_dir)


# dataroot = '/data/ziqiang/Benchmark'
# dataroot = '/data/ziqiang/jpeg'
# dataroot = '/data/ziqiang/noise'
dataroot = '/data/ziqiang/sample'


vals = [
    'BlendFace',
    'BLIP',
    'CommunityAI',
    'DALLE-3',
    'E4S',
    'FaceSwap',
    'FLUX1-dev',
    'GLIDE',
    'Imagen3',
    'Infinite-ID',
    'InstantID',
    'InSwap',
    'IP-Adapter',
    'Midjourney',
    'PhotoMaker',
    'ProGAN',
    'R3GAN',
    'SD3',
    'SDXL',
    'SimSwap',
    'SocialRF',
    'StyleGAN-XL',
    'StyleGAN3',
    'StyleSwim',
    'WFIR',
]
multiclass = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

# model
model_path = '/data/ziqiang/yjz/code/CNNDetection-master/checkpoints/4class-car-cat-chair-horse2025_04_12_14_39_53/model_epoch_best.pth'