Skip to content
Snippets Groups Projects
Commit ef97e51a authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[database] fixed high-level interface for CASIA-SURF

parent 6da741c1
No related branches found
No related tags found
1 merge request!77CASIA-SURF database
Pipeline #26254 passed
...@@ -46,12 +46,11 @@ class CasiaSurfPadFile(VideoPadFile): ...@@ -46,12 +46,11 @@ class CasiaSurfPadFile(VideoPadFile):
client_id=s.id, client_id=s.id,
file_id=s.id, file_id=s.id,
attack_type=attack_type, attack_type=attack_type,
path='') path=s.id)
def load(self, directory=None, extension='.jpg', frame_selector=FrameSelector(selection_style='all')): def load(self, directory=None, extension='.jpg', frame_selector=FrameSelector(selection_style='all')):
"""Overridden version of the load method defined in ``VideoPadFile``. """Overloaded version of the load method defined in ``VideoPadFile``.
Parameters Parameters
---------- ----------
...@@ -165,9 +164,7 @@ class CasiaSurfPadDatabase(PadDatabase): ...@@ -165,9 +164,7 @@ class CasiaSurfPadDatabase(PadDatabase):
A list of CasiaSurfPadFile objects. A list of CasiaSurfPadFile objects.
""" """
print(groups)
groups = self.convert_names_to_lowlevel(groups, self.low_level_group_names, self.high_level_group_names) groups = self.convert_names_to_lowlevel(groups, self.low_level_group_names, self.high_level_group_names)
print(groups)
if groups is not None: if groups is not None:
...@@ -179,17 +176,11 @@ class CasiaSurfPadDatabase(PadDatabase): ...@@ -179,17 +176,11 @@ class CasiaSurfPadDatabase(PadDatabase):
lowlevel_purposes.append('attack') lowlevel_purposes.append('attack')
# for dev and eval # for dev and eval
if ('validation' in groups or 'test' in groups) and purposes == 'real': if ('validation' in groups or 'test' in groups) and 'attack' in purposes:
lowlevel_purposes.append('unknown')
if ('validation' in groups or 'test' in groups) and purposes == 'attack':
lowlevel_purposes.append('unknown') lowlevel_purposes.append('unknown')
print("In high-level DB: groups = {}".format(groups))
print("In high-level DB: purposes = {}".format(lowlevel_purposes))
print(lowlevel_purposes)
samples = self.db.objects(groups=groups, purposes=lowlevel_purposes, **kwargs) samples = self.db.objects(groups=groups, purposes=lowlevel_purposes, **kwargs)
samples = [CasiaSurfPadFile(s, stream_type=protocol) for s in samples] samples = [CasiaSurfPadFile(s, stream_type=protocol) for s in samples]
return samples return samples
...@@ -197,4 +188,3 @@ class CasiaSurfPadDatabase(PadDatabase): ...@@ -197,4 +188,3 @@ class CasiaSurfPadDatabase(PadDatabase):
"""No annotations are provided with this DB """No annotations are provided with this DB
""" """
return None return None
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