Skip to content
Snippets Groups Projects
Commit 0412ad80 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Patch dbs

parent 0785fcd7
No related branches found
No related tags found
1 merge request!112Feature extractors
Pipeline #52233 passed
......@@ -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():
......
......@@ -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():
......
......@@ -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():
......
......@@ -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():
......
......@@ -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():
......
......@@ -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():
......
......@@ -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()
......
......@@ -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():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment