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