| TASK: heatmap | |
| PRECISION: float32 | |
| DATASET: | |
| type: HeatmapFaceForensic | |
| TRAIN: True | |
| ROOT: /data/deepfake_cluster/datasets_df/FaceForensics++/c23/ | |
| # FAKETYPE: [Deepfakes, Face2Face, FaceSwap, NeuralTextures] | |
| FAKETYPE: [FaceXRay] | |
| IMAGE_SUFFIX: jpg | |
| FROM_FILE: True | |
| NUM_WORKERS: 8 | |
| PIN_MEMORY: True | |
| IMAGE_SIZE: [256, 256] | |
| HEATMAP_SIZE: [64, 64] | |
| SIGMA: 2 | |
| HEATMAP_TYPE: gaussian | |
| DATA: | |
| TYPE: images | |
| TRAIN: | |
| ANNO_FILE: FaceXRay/train/train_FF_Xray.json | |
| VAL: | |
| ANNO_FILE: FaceXRay/val/val_FF_Xray.json | |
| TEST: | |
| ANNO_FILE: FaceXRay/test/test_FF_Xray.json | |
| TRANSFORM: | |
| geometry: | |
| type: GeometryTransform | |
| resize: [256, 256, 0] #h, w, p=probability. If no affine transform, set p=1 | |
| color: | |
| type: ColorJitterTransform | |
| clahe: 0.5 | |
| colorjitter: 0.5 | |
| gaussianblur: 0.5 | |
| jpegcompression: 0.5 | |
| rgbshift: 0.5 | |
| normalize: | |
| mean: [0.485, 0.456, 0.406] | |
| std: [0.229, 0.224, 0.225] | |
| DEBUG: False | |
| MODEL: | |
| type: SimpleClassificationDF | |
| backbone: | |
| type: ResNet | |
| num_layers: 50 | |
| drop_ratio: 0.5 | |
| mode: ir_se | |
| head: | |
| type: SimpleClassificationHead | |
| drop_ratio: 0.5 | |
| in_planes: 512 | |
| TRAIN: | |
| gpus: [0,1,2,3] | |
| batch_size: 32 | |
| lr: 0.001 | |
| epochs: 100 | |
| begin_epoch: 0 | |
| warm_up: 5 | |
| every_val_epochs: 3 | |
| loss: | |
| type: CombinedLoss | |
| use_target_weight: False | |
| optimizer: Adam | |
| distributed: False | |
| pretrained: pretrained/model_ir_se50.pth | |
| tensorboard: True | |
| resume: False | |
| lr_scheduler: | |
| # type: MultiStepLR | |
| milestones: [50, 80, 90] | |
| gamma: 0.1 | |
| PREPROCESSING: | |
| DATASET: FaceForensics | |
| SPLIT: train | |
| ROOT: /data/deepfake_cluster/datasets_df/FaceForensics++/c23/ | |
| FAKETYPE: [Deepfakes, Face2Face, FaceSwap, NeuralTextures] | |
| IMAGE_SUFFIX: jpg | |
| DATA_TYPE: images | |
| LABEL: real | |
| facial_lm_pretrained: pretrained/shape_predictor_68_face_landmarks.dat | |