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