From b3cbbc5111732b51f73482732de1725ef3ed8bd6 Mon Sep 17 00:00:00 2001 From: Yannick DAYER <yannick.dayer@idiap.ch> Date: Wed, 15 Feb 2023 02:37:21 +0100 Subject: [PATCH] [py] Add database names required by bob.pipelines. --- src/bob/pad/face/database/casia_fasd.py | 3 ++- src/bob/pad/face/database/casia_surf.py | 1 + src/bob/pad/face/database/mask_attack.py | 5 +++-- src/bob/pad/face/database/oulu_npu.py | 5 +++-- src/bob/pad/face/database/replay_attack.py | 5 +++-- src/bob/pad/face/database/replay_mobile.py | 5 +++-- 6 files changed, 15 insertions(+), 9 deletions(-) diff --git a/src/bob/pad/face/database/casia_fasd.py b/src/bob/pad/face/database/casia_fasd.py index f6508fd7..625e78af 100644 --- a/src/bob/pad/face/database/casia_fasd.py +++ b/src/bob/pad/face/database/casia_fasd.py @@ -36,7 +36,8 @@ def CasiaFasdPadDatabase( ) database = FileListPadDatabase( - dataset_protocols_path, + name="casia-fsd", + dataset_protocols_path=dataset_protocols_path, protocol="grandtest", transformer=transformer, **kwargs, diff --git a/src/bob/pad/face/database/casia_surf.py b/src/bob/pad/face/database/casia_surf.py index 4795d8a9..7df38956 100644 --- a/src/bob/pad/face/database/casia_surf.py +++ b/src/bob/pad/face/database/casia_surf.py @@ -89,6 +89,7 @@ class CasiaSurfPadDatabase(FileListPadDatabase): original_directory=original_directory, ) super().__init__( + name="casia-surf", dataset_protocols_path=original_directory, protocol="all", reader_cls=partial( diff --git a/src/bob/pad/face/database/mask_attack.py b/src/bob/pad/face/database/mask_attack.py index 04a487ea..57fd702f 100644 --- a/src/bob/pad/face/database/mask_attack.py +++ b/src/bob/pad/face/database/mask_attack.py @@ -144,8 +144,9 @@ def MaskAttackPadDatabase( ) database = FileListPadDatabase( - dataset_protocols_path, - protocol, + name="mask-attack", + dataset_protocols_path=dataset_protocols_path, + protocol=protocol, transformer=transformer, **kwargs, ) diff --git a/src/bob/pad/face/database/oulu_npu.py b/src/bob/pad/face/database/oulu_npu.py index fec11f76..d3cdebbf 100644 --- a/src/bob/pad/face/database/oulu_npu.py +++ b/src/bob/pad/face/database/oulu_npu.py @@ -47,8 +47,9 @@ def OuluNpuPadDatabase( ) database = FileListPadDatabase( - dataset_protocols_path, - protocol, + name="oulunpu", + dataset_protocols_path=dataset_protocols_path, + protocol=protocol, transformer=transformer, **kwargs, ) diff --git a/src/bob/pad/face/database/replay_attack.py b/src/bob/pad/face/database/replay_attack.py index c9a2a69f..31e55907 100644 --- a/src/bob/pad/face/database/replay_attack.py +++ b/src/bob/pad/face/database/replay_attack.py @@ -47,8 +47,9 @@ def ReplayAttackPadDatabase( ) database = FileListPadDatabase( - dataset_protocols_path, - protocol, + name="replay-attack", + dataset_protocols_path=dataset_protocols_path, + protocol=protocol, transformer=transformer, **kwargs, ) diff --git a/src/bob/pad/face/database/replay_mobile.py b/src/bob/pad/face/database/replay_mobile.py index fb23c862..4a6db65a 100644 --- a/src/bob/pad/face/database/replay_mobile.py +++ b/src/bob/pad/face/database/replay_mobile.py @@ -65,8 +65,9 @@ def ReplayMobilePadDatabase( ) database = FileListPadDatabase( - dataset_protocols_path, - protocol, + name="replay-mobile", + dataset_protocols_path=dataset_protocols_path, + protocol=protocol, transformer=transformer, **kwargs, ) -- GitLab