TheKernel01's picture
Sync from GitHub via hub-sync
0788e19 verified
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'