clean/image/yermandy/src/utils/files.py

def get_FF_train_files():
    return [
        "config/datasets/FF/train/DF.txt",
        "config/datasets/FF/train/F2F.txt",
        "config/datasets/FF/train/FS.txt",
        "config/datasets/FF/train/NT.txt",
        "config/datasets/FF/train/real.txt",
    ]


def get_FF_val_files():
    return [
        "config/datasets/FF/val/DF.txt",
        "config/datasets/FF/val/F2F.txt",
        "config/datasets/FF/val/FS.txt",
        "config/datasets/FF/val/NT.txt",
        "config/datasets/FF/val/real.txt",
    ]


def get_FF_test_files():
    return [
        "config/datasets/FF/test/DF.txt",
        "config/datasets/FF/test/F2F.txt",
        "config/datasets/FF/test/FS.txt",
        "config/datasets/FF/test/NT.txt",
        "config/datasets/FF/test/real.txt",
    ]


def get_FF_DF_test_files():
    return [
        "config/datasets/FF/test/DF.txt",
        "config/datasets/FF/test/real.txt",
    ]


def get_FF_F2F_test_files():
    return [
        "config/datasets/FF/test/F2F.txt",
        "config/datasets/FF/test/real.txt",
    ]


def get_FF_FS_test_files():
    return [
        "config/datasets/FF/test/FS.txt",
        "config/datasets/FF/test/real.txt",
    ]


def get_FF_NT_test_files():
    return [
        "config/datasets/FF/test/NT.txt",
        "config/datasets/FF/test/real.txt",
    ]


def get_CDF_all_files():
    return [
        "config/datasets/CDFv2/all/Celeb-synthesis.txt",
        "config/datasets/CDFv2/all/YouTube-real.txt",
        "config/datasets/CDFv2/all/Celeb-real.txt",
    ]


def get_CDF_test_files():
    return [
        "config/datasets/CDFv2/test/Celeb-synthesis.txt",
        "config/datasets/CDFv2/test/YouTube-real.txt",
        "config/datasets/CDFv2/test/Celeb-real.txt",
    ]


def get_FF_FaceFusion_train_files():
    return [
        "config/datasets/FaceFusion/train/ff_inswapper_128_fp16.txt",
    ]


def get_CDF_FaceFusion_test_files():
    return [
        "config/datasets/FaceFusion/test/cdf_hififace_unofficial_256.txt",
        "config/datasets/FaceFusion/test/cdf_inswapper_128_fp16.txt",
        "config/datasets/CDFv2/test/YouTube-real.txt",
        "config/datasets/CDFv2/test/Celeb-real.txt",
    ]


def get_DFD_files():
    return [
        "config/datasets/DFD/fake.txt",
        "config/datasets/DFD/real.txt",
    ]


def get_DFDC_test_files():
    return [
        "config/datasets/DFDC/test/fake.txt",
        "config/datasets/DFDC/test/real.txt",
    ]


def get_FSh_test_files():
    return [
        "config/datasets/FSh/test/fake.txt",
        "config/datasets/FSh/test/real.txt",
    ]


def get_UADFD_files():
    return [
        "config/datasets/UADFD/fake.txt",
        "config/datasets/UADFD/real.txt",
    ]


def get_DFDM_files():
    return [
        "config/datasets/DFDM/all/dfaker.txt",
        "config/datasets/DFDM/all/dfl.txt",
        "config/datasets/DFDM/all/iae.txt",
        "config/datasets/DFDM/all/lightweight.txt",
        "config/datasets/CDFv2/all/Celeb-real.txt",
    ]


def get_FFIW_files():
    return [
        "config/datasets/FFIW/fake.txt",
        "config/datasets/FFIW/real.txt",
    ]


def get_AVLips_files():
    return [
        "config/datasets/AVLips/fake.txt",
        "config/datasets/AVLips/real.txt",
    ]


def get_DeepSpeak_test_files():
    return [
        "config/datasets/DeepSpeak/test-facefusion_gan.txt",
        "config/datasets/DeepSpeak/test-facefusion_live.txt",
        "config/datasets/DeepSpeak/test-facefusion.txt",
        "config/datasets/DeepSpeak/test-real.txt",
        "config/datasets/DeepSpeak/test-retalking.txt",
        "config/datasets/DeepSpeak/test-wav2lip.txt",
    ]


def get_DeepSpeak_train_files():
    return [
        "config/datasets/DeepSpeak/train-facefusion_gan.txt",
        "config/datasets/DeepSpeak/train-facefusion_live.txt",
        "config/datasets/DeepSpeak/train-facefusion.txt",
        "config/datasets/DeepSpeak/train-real.txt",
        "config/datasets/DeepSpeak/train-retalking.txt",
        "config/datasets/DeepSpeak/train-wav2lip.txt",
    ]