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):
The protocol for which the samples should be retrieved.
purposes : :py:class:`str`
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
This parameter is not supported in PAD databases yet.
......@@ -160,9 +160,9 @@ class CasiaSurfPadDatabase(PadDatabase):
# for training
lowlevel_purposes = []
if 'train' in groups and purposes == 'real':
if 'train' in groups and 'real' in purposes:
lowlevel_purposes.append('real')
if 'train' in groups and purposes == 'attack':
if 'train' in groups and 'attack' in purposes:
lowlevel_purposes.append('attack')
# 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