From 6da741c18e0c21c03155aaa6a86e60f747f2e2a9 Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Thu, 17 Jan 2019 10:49:07 +0100
Subject: [PATCH] experimenting with the database ...
---
bob/pad/face/database/casiasurf.py | 23 +++++++++++++++++------
1 file changed, 17 insertions(+), 6 deletions(-)
diff --git a/bob/pad/face/database/casiasurf.py b/bob/pad/face/database/casiasurf.py
index a77b3ee2..b58b0a7e 100644
--- a/bob/pad/face/database/casiasurf.py
+++ b/bob/pad/face/database/casiasurf.py
@@ -35,14 +35,20 @@ class CasiaSurfPadFile(VideoPadFile):
stream_type: str of list of str
The streams to be loaded.
"""
- self.path = None
self.s = s
self.stream_type = stream_type
+ if not isinstance(s.attack_type, str):
+ attack_type = str(s.attack_type)
+ else:
+ attack_type = s.attack_type
+
super(CasiaSurfPadFile, self).__init__(
client_id=s.id,
- attack_type=s.attack_type,
- path=None)
+ file_id=s.id,
+ attack_type=attack_type,
+ path='')
+
def load(self, directory=None, extension='.jpg', frame_selector=FrameSelector(selection_style='all')):
"""Overridden version of the load method defined in ``VideoPadFile``.
@@ -65,7 +71,7 @@ class CasiaSurfPadFile(VideoPadFile):
"""
# get the dict of numpy array
- data = self.f.load(directory, extension, modality=self.stream_type)
+ data = self.s.load(directory, extension, modality=self.stream_type)
# convert that to dict of FrameContainer
data_to_return = {}
@@ -159,7 +165,9 @@ class CasiaSurfPadDatabase(PadDatabase):
A list of CasiaSurfPadFile objects.
"""
+ print(groups)
groups = self.convert_names_to_lowlevel(groups, self.low_level_group_names, self.high_level_group_names)
+ print(groups)
if groups is not None:
@@ -171,11 +179,14 @@ class CasiaSurfPadDatabase(PadDatabase):
lowlevel_purposes.append('attack')
# for dev and eval
- if ('dev' in groups or 'test' in groups) and purposes == 'real':
+ if ('validation' in groups or 'test' in groups) and purposes == 'real':
lowlevel_purposes.append('unknown')
- if ('dev' in groups or 'test' in groups) and purposes == 'attack':
+ if ('validation' in groups or 'test' in groups) and purposes == 'attack':
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 = [CasiaSurfPadFile(s, stream_type=protocol) for s in samples]
--
GitLab