diff --git a/src/generated_image_detection/constants.py b/src/generated_image_detection/constants.py index 435d9121b863e6e4115901a9f2c1418493655051..0d8c66936c06913ac51fe7f5ca0f38d2dc10165a 100644 --- a/src/generated_image_detection/constants.py +++ b/src/generated_image_detection/constants.py @@ -16,7 +16,7 @@ from generated_image_detection.transformations.transformation_factory import Ava DatasetType = Literal[ "Real", "Fake", - # "Mixed", # TODO: not handled for now + "Mixed", # (Needed for DeepfakeEcal2024) ] FakeDatasets = Literal[ "AISummitCustom", @@ -31,6 +31,8 @@ FakeDatasets = Literal[ "FaceSwap", "NeuralTextures", "SFHQ_T2I", + "AMMeBa", + "GPT-ImgEval", ] RealDatasets = Literal[ "AFHQ2", @@ -44,8 +46,13 @@ RealDatasets = Literal[ "LSUN", "Metfaces", "UCID", + "Artchive", ] -ImageDataset = Literal[FakeDatasets, RealDatasets, "Custom"] + +MixedDatasets = Literal["DeepfakeEval2024"] + + +ImageDataset = Literal[FakeDatasets, RealDatasets, MixedDatasets, "Custom"] GenerativeArchitecture = Literal["GAN", "Diffusion", "Other", "Real"]