Patch dbs

parent 0785fcd7
Pipeline #52233 passed with stage
in 42 minutes and 27 seconds
......@@ -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():
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment