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

Updated database interface

parent 30ad7b48
No related branches found
No related tags found
1 merge request!112Feature extractors
Pipeline #51310 passed
......@@ -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,
......
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