diff --git a/bob/pad/face/database/__init__.py b/bob/pad/face/database/__init__.py index 12814bda6532ef69b43c9f4ac5425c6001f2d8ba..87a1efccb00a2504bad0e73c99c6b17519272d1b 100644 --- a/bob/pad/face/database/__init__.py +++ b/bob/pad/face/database/__init__.py @@ -4,6 +4,7 @@ from .replay_mobile import ReplayMobilePadDatabase from .msu_mfsd import MsuMfsdPadDatabase from .aggregated_db import AggregatedDbPadDatabase from .mifs import MIFSPadDatabase +from .batl import BatlPadDatabase # gets sphinx autodoc done right - don't remove it @@ -29,6 +30,7 @@ __appropriate__( MsuMfsdPadDatabase, AggregatedDbPadDatabase, MIFSPadDatabase, + BatlPadDatabase, ) __all__ = [_ for _ in dir() if not _.startswith('_')] diff --git a/bob/pad/face/database/batl.py b/bob/pad/face/database/batl.py index ff66c799abc6ab42797686f0613e1cef2da18f6b..0a1c90fe77c41e4e24b7bf1b4c69d40a311b4336 100644 --- a/bob/pad/face/database/batl.py +++ b/bob/pad/face/database/batl.py @@ -192,7 +192,7 @@ class BatlPadDatabase(PadDatabase): def objects(self, protocol=None, - group=None, + groups=None, purposes=None, sessions=None, **kwargs): @@ -229,6 +229,26 @@ class BatlPadDatabase(PadDatabase): # Since this database was designed for PAD experiments, nothing special # needs to be done here. files = self.db.objects(protocol=protocol, groups=groups, purposes=purposes **kwargs) + + +# files = self.db.objects(protocol=protocol, purposes=groups, **kwargs) +# +# if purposes == ["real", "attack"]: +# +# files = files +# +# if purposes == "real" or purposes == ["real"]: +# +# a = 1 +# +# +# if purposes == "attack" or purposes == ["attack"]: +# +# a = 1 + + + + files = [BatlPadFile(f, stream_type, max_frames) for f in files] return files