Commit ba0468cc authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Updated database interface

parent 30ad7b48
Pipeline #51310 passed with stage
in 36 minutes and 37 seconds
......@@ -117,8 +117,9 @@ class CasiaAfricaDatabase(CSVDataset):
)
super().__init__(
filename,
protocol,
name="casia-africa",
dataset_protocol_path=filename,
protocol=protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=bob.io.base.load,
......
......@@ -81,8 +81,9 @@ class CBSRNirVis2Database(CSVDataset):
raise ValueError("File `{0}` not found".format(str(new_filename)))
super().__init__(
filename,
protocol,
name="cbsr-nir-vis2",
dataset_protocol_path=filename,
protocol=protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=load,
......
......@@ -32,8 +32,9 @@ class FRGCDatabase(CSVDataset):
self.fixed_positions = None
super().__init__(
filename,
protocol,
name="frgc",
dataset_protocol_path=filename,
protocol=protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=bob.io.base.load,
......
......@@ -104,9 +104,10 @@ class MEDSDatabase(CSVDatasetZTNorm):
self.annotation_type = "eyes-center"
self.fixed_positions = None
database = CSVDataset(
filename,
protocol,
super().__init__(
name="meds",
dataset_protocol_path=filename,
protocol=protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=bob.io.base.load,
......@@ -119,8 +120,6 @@ class MEDSDatabase(CSVDatasetZTNorm):
),
)
super().__init__(database)
@staticmethod
def urls():
return [
......
......@@ -59,9 +59,10 @@ class MobioDatabase(CSVDatasetZTNorm):
self.annotation_type = "eyes-center"
self.fixed_positions = None
database = CSVDataset(
filename,
protocol,
super().__init__(
name="mobio",
dataset_protocol_path=filename,
protocol=protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=bob.io.base.load,
......@@ -74,8 +75,6 @@ class MobioDatabase(CSVDatasetZTNorm):
),
)
super().__init__(database)
@staticmethod
def protocols():
# TODO: Until we have (if we have) a function that dumps the protocols, let's use this one.
......
......@@ -68,9 +68,10 @@ class MorphDatabase(CSVDatasetZTNorm):
self.annotation_type = "eyes-center"
self.fixed_positions = None
database = CSVDataset(
filename,
protocol,
super().__init__(
name="morph",
dataset_protocol_path=filename,
protocol=protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=bob.io.base.load,
......@@ -83,8 +84,6 @@ class MorphDatabase(CSVDatasetZTNorm):
),
)
super().__init__(database)
@staticmethod
def urls():
return [
......
......@@ -98,8 +98,9 @@ class MultipieDatabase(CSVDataset):
self.fixed_positions = None
super().__init__(
filename,
protocol,
name="multipie",
dataset_protocol_path=filename,
protocol=protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=bob.io.base.load,
......
......@@ -103,8 +103,9 @@ class PolaThermalDatabase(CSVDataset):
return bob.io.base.load(path) / 255
super().__init__(
filename,
protocol,
name="polathermal",
protocol=protocol,
dataset_protocol_path=filename,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=load,
......
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