Spaces:
Sleeping
Sleeping
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'
|