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

[database] small fix in CASIA-SURF

parent de183c93
No related branches found
No related tags found
1 merge request!87Chalearn pad
Pipeline #26876 passed
...@@ -144,7 +144,7 @@ class CasiaSurfPadDatabase(PadDatabase): ...@@ -144,7 +144,7 @@ class CasiaSurfPadDatabase(PadDatabase):
The protocol for which the samples should be retrieved. The protocol for which the samples should be retrieved.
purposes : :py:class:`str` purposes : :py:class:`str`
The purposes for which Sample objects should be retrieved. The purposes for which Sample objects should be retrieved.
Usually it is either 'real' or 'attack', but could be 'unknown' as well Usually it is either 'real' or 'attack'
model_ids model_ids
This parameter is not supported in PAD databases yet. This parameter is not supported in PAD databases yet.
...@@ -160,9 +160,9 @@ class CasiaSurfPadDatabase(PadDatabase): ...@@ -160,9 +160,9 @@ class CasiaSurfPadDatabase(PadDatabase):
# for training # for training
lowlevel_purposes = [] lowlevel_purposes = []
if 'train' in groups and purposes == 'real': if 'train' in groups and 'real' in purposes:
lowlevel_purposes.append('real') lowlevel_purposes.append('real')
if 'train' in groups and purposes == 'attack': if 'train' in groups and 'attack' in purposes:
lowlevel_purposes.append('attack') lowlevel_purposes.append('attack')
# for dev and eval # for dev and eval
......
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