diff --git a/bob/bio/face/database/casia_africa.py b/bob/bio/face/database/casia_africa.py index bfedd075b0da1b9612d6db04cbb0b2a28c05fe32..34dc60f89d86f7ca27e6a5bdc25bcffdc720821c 100644 --- a/bob/bio/face/database/casia_africa.py +++ b/bob/bio/face/database/casia_africa.py @@ -97,7 +97,7 @@ class CasiaAfricaDatabase(CSVDataset): One of the database protocols. Options are "ID-V-All-Ep1", "ID-V-All-Ep2" and "ID-V-All-Ep3" """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = CasiaAfricaDatabase.urls() @@ -128,8 +128,8 @@ class CasiaAfricaDatabase(CSVDataset): ), ) - self.annotation_type = "eyes-center" - self.fixed_positions = None + self.annotation_type = annotation_type + self.fixed_positions = fixed_positions @staticmethod def protocols(): diff --git a/bob/bio/face/database/cbsr_nir_vis_2.py b/bob/bio/face/database/cbsr_nir_vis_2.py index c6ec79f16df82af35ebd67222633692b2d96c751..bab3a4f72a47395707a97e526ce10a578c68732a 100644 --- a/bob/bio/face/database/cbsr_nir_vis_2.py +++ b/bob/bio/face/database/cbsr_nir_vis_2.py @@ -54,7 +54,7 @@ class CBSRNirVis2Database(CSVDataset): One of the database protocols. """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = CBSRNirVis2Database.urls() @@ -91,8 +91,8 @@ class CBSRNirVis2Database(CSVDataset): EyesAnnotations(), ), ) - self.annotation_type = "eyes-center" - self.fixed_positions = None + self.annotation_type = annotation_type + self.fixed_positions = fixed_positions @staticmethod def protocols(): diff --git a/bob/bio/face/database/frgc.py b/bob/bio/face/database/frgc.py index 38de5875b6c8c52dda14b8c37f125d72cedca5f5..a2831f6e78cd7fd95daa5b7ec5ebe5588c4ef50c 100644 --- a/bob/bio/face/database/frgc.py +++ b/bob/bio/face/database/frgc.py @@ -20,7 +20,7 @@ class FRGCDatabase(CSVDataset): Face Recognition Grand Test dataset """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = FRGCDatabase.urls() @@ -44,8 +44,8 @@ class FRGCDatabase(CSVDataset): ), ) - self.annotation_type = "eyes-center" - self.fixed_positions = None + self.annotation_type = annotation_type + self.fixed_positions = fixed_positions @staticmethod def protocols(): diff --git a/bob/bio/face/database/meds.py b/bob/bio/face/database/meds.py index 18101472bf63e9a414799f00a3291c93dbc48ab8..397cc69e742980c7228c05210932d402f2a1908e 100644 --- a/bob/bio/face/database/meds.py +++ b/bob/bio/face/database/meds.py @@ -93,7 +93,7 @@ class MEDSDatabase(CSVDatasetZTNorm): """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = MEDSDatabase.urls() @@ -117,8 +117,8 @@ class MEDSDatabase(CSVDatasetZTNorm): ), ) - self.annotation_type = "eyes-center" - self.fixed_positions = None + self.annotation_type = annotation_type + self.fixed_positions = fixed_positions @staticmethod def urls(): diff --git a/bob/bio/face/database/mobio.py b/bob/bio/face/database/mobio.py index 02af60a2cb1978524bc9aac60d0dc11e17770f7d..729e9a8862bad196db6b9f053fcaad2ea9d629db 100644 --- a/bob/bio/face/database/mobio.py +++ b/bob/bio/face/database/mobio.py @@ -48,7 +48,7 @@ class MobioDatabase(CSVDatasetZTNorm): """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = MobioDatabase.urls() @@ -71,8 +71,8 @@ class MobioDatabase(CSVDatasetZTNorm): EyesAnnotations(), ), ) - self.annotation_type = "eyes-center" - self.fixed_positions = None + self.annotation_type = annotation_type + self.fixed_positions = fixed_positions @staticmethod def protocols(): diff --git a/bob/bio/face/database/morph.py b/bob/bio/face/database/morph.py index be557b02c2f9d95cf4c30ad86ea4200386e5b411..391292ddd4ca15d84f359c937d8362a86e4a1e3e 100644 --- a/bob/bio/face/database/morph.py +++ b/bob/bio/face/database/morph.py @@ -57,7 +57,7 @@ class MorphDatabase(CSVDatasetZTNorm): """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = MorphDatabase.urls() @@ -81,8 +81,8 @@ class MorphDatabase(CSVDatasetZTNorm): ), ) - self.annotation_type = "eyes-center" - self.fixed_positions = None + self.annotation_type = annotation_type + self.fixed_positions = fixed_positions @staticmethod def urls(): diff --git a/bob/bio/face/database/multipie.py b/bob/bio/face/database/multipie.py index 572ca1e929c57631a244706de66a74bfc3e82b86..6bfed594ff9552840adb88f4e1b9365d9f9fcfed 100644 --- a/bob/bio/face/database/multipie.py +++ b/bob/bio/face/database/multipie.py @@ -86,7 +86,7 @@ class MultipieDatabase(CSVDataset): """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = MultipieDatabase.urls() diff --git a/bob/bio/face/database/pola_thermal.py b/bob/bio/face/database/pola_thermal.py index 1a976de90d1d452b62f0e549ca60c6eff57e3a2b..cb1808a0050526b7af688e399782d0f136af5982 100644 --- a/bob/bio/face/database/pola_thermal.py +++ b/bob/bio/face/database/pola_thermal.py @@ -79,7 +79,7 @@ class PolaThermalDatabase(CSVDataset): One of the database protocols. """ - def __init__(self, protocol): + def __init__(self, protocol, annotation_type="eyes-center", fixed_positions=None): # Downloading model if not exists urls = PolaThermalDatabase.urls() @@ -113,8 +113,8 @@ class PolaThermalDatabase(CSVDataset): ), ) - self.annotation_type = "eyes-center" - self.fixed_positions = None + self.annotation_type = annotation_type + self.fixed_positions = fixed_positions @staticmethod def protocols():