Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
bob
bob.pad.face
Commits
8cd56b44
Commit
8cd56b44
authored
Jan 31, 2019
by
Anjith GEORGE
Browse files
Added unit tests for protocols in BATL HLDI
parent
178e48d3
Pipeline
#26473
failed with stage
in 8 minutes and 45 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
bob/pad/face/test/test_databases.py
View file @
8cd56b44
...
...
@@ -223,7 +223,7 @@ def test_casiasurf():
# Test the BATL database
def
test_aggregated_db
():
batl_db
=
bob
.
bio
.
base
.
load_resource
(
'batl'
,
'batl
-db
'
,
'database'
,
preferred_package
=
'bob.pad.face'
,
package_prefix
=
'bob.pad.'
)
...
...
@@ -274,6 +274,78 @@ def test_aggregated_db():
batl_db
.
objects
(
groups
=
'eval'
,
protocol
=
'grandtest-color-50-LOO_fakehead'
))
==
132
# test for LOO_flexiblemask
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
,
'eval'
],
protocol
=
'grandtest-color-50-LOO_flexiblemask'
))
==
1132
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
],
protocol
=
'grandtest-color-50-LOO_flexiblemask'
))
==
880
assert
len
(
batl_db
.
objects
(
groups
=
'eval'
,
protocol
=
'grandtest-color-50-LOO_flexiblemask'
))
==
252
# test for LOO_glasses
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
,
'eval'
],
protocol
=
'grandtest-color-50-LOO_glasses'
))
==
1206
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
],
protocol
=
'grandtest-color-50-LOO_glasses'
))
==
1069
assert
len
(
batl_db
.
objects
(
groups
=
'eval'
,
protocol
=
'grandtest-color-50-LOO_glasses'
))
==
137
# test for LOO_papermask
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
,
'eval'
],
protocol
=
'grandtest-color-50-LOO_papermask'
))
==
1308
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
],
protocol
=
'grandtest-color-50-LOO_papermask'
))
==
1122
assert
len
(
batl_db
.
objects
(
groups
=
'eval'
,
protocol
=
'grandtest-color-50-LOO_papermask'
))
==
186
# test for LOO_prints
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
,
'eval'
],
protocol
=
'grandtest-color-50-LOO_prints'
))
==
1169
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
],
protocol
=
'grandtest-color-50-LOO_prints'
))
==
988
assert
len
(
batl_db
.
objects
(
groups
=
'eval'
,
protocol
=
'grandtest-color-50-LOO_prints'
))
==
181
# test for LOO_replay
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
,
'eval'
],
protocol
=
'grandtest-color-50-LOO_replay'
))
==
1049
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
],
protocol
=
'grandtest-color-50-LOO_replay'
))
==
854
assert
len
(
batl_db
.
objects
(
groups
=
'eval'
,
protocol
=
'grandtest-color-50-LOO_replay'
))
==
195
# test for LOO_rigidmask
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
,
'eval'
],
protocol
=
'grandtest-color-50-LOO_rigidmask'
))
==
1198
assert
len
(
batl_db
.
objects
(
groups
=
[
'train'
,
'dev'
],
protocol
=
'grandtest-color-50-LOO_rigidmask'
))
==
1034
assert
len
(
batl_db
.
objects
(
groups
=
'eval'
,
protocol
=
'grandtest-color-50-LOO_rigidmask'
))
==
164
except
IOError
as
e
:
raise
SkipTest
(
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment