Skip to content
Snippets Groups Projects
Commit b4486899 authored by Yannick DAYER's avatar Yannick DAYER
Browse files

Merge branch 'fix-db-metadata' into 'master'

fix [database]: Add template metadata to databases

See merge request !207
parents c6b6de0d e86d4d38
No related branches found
No related tags found
1 merge request!207fix [database]: Add template metadata to databases
Pipeline #73338 passed
......@@ -119,6 +119,7 @@ class MEDSDatabase(CSVDatabase):
),
EyesAnnotations(),
),
templates_metadata=["sex", "race"],
annotation_type=annotation_type,
fixed_positions=fixed_positions,
)
......@@ -85,6 +85,7 @@ class MobioDatabase(CSVDatabase):
),
EyesAnnotations(),
),
templates_metadata=["gender"],
annotation_type=annotation_type,
fixed_positions=fixed_positions,
)
......@@ -86,6 +86,7 @@ class MorphDatabase(CSVDatabase):
),
EyesAnnotations(),
),
templates_metadata=["date_of_birth", "sex", "rac"],
annotation_type=annotation_type,
fixed_positions=fixed_positions,
)
......@@ -191,6 +191,7 @@ class VGG2Database(CSVDatabase):
),
VGG2Annotations(),
),
templates_metadata=["gender", "race"],
annotation_type=annotation_type,
fixed_positions=fixed_positions,
)
......
......@@ -93,6 +93,8 @@ def test_mobio():
assert len(database.references(group="eval")) > 0
assert len(database.probes(group="eval")) > 0
assert hasattr(database.references()[0], "gender")
# Sanity check on mobio-male
database = MobioDatabase(protocol="mobile0-male")
assert len(database.treferences()) == 24
......@@ -275,6 +277,11 @@ def test_meds():
assert len(database.references(group="eval")) == 112
assert len(database.probes(group="eval")) == 309
assert hasattr(database.references()[0], "race")
assert hasattr(database.references()[0], "sex")
assert hasattr(database.probes()[0], "race")
assert hasattr(database.probes()[0], "sex")
def test_morph():
from bob.bio.face.database import MorphDatabase
......@@ -303,6 +310,13 @@ def test_morph():
assert len(database.references(group="eval")) == 6742
assert len(database.probes(group="eval")) == 6553
assert hasattr(database.references()[0], "date_of_birth")
assert hasattr(database.references()[0], "rac")
assert hasattr(database.references()[0], "sex")
assert hasattr(database.probes()[0], "date_of_birth")
assert hasattr(database.probes()[0], "rac")
assert hasattr(database.probes()[0], "sex")
def test_casia_africa():
from bob.bio.face.database import CasiaAfricaDatabase
......@@ -394,6 +408,11 @@ def test_rfw():
assert len(database.treferences()) == 100
assert sum([len(d.references) for d in database.probes()]) == 89540
assert hasattr(database.references()[0], "gender")
assert hasattr(database.references()[0], "race")
assert hasattr(database.probes()[0], "gender")
assert hasattr(database.probes()[0], "race")
def test_scface():
from bob.bio.face.database import SCFaceDatabase
......@@ -750,6 +769,11 @@ def test_vgg2():
assert len(database.references()) == 500
assert len(database.probes()) == 2500
assert hasattr(database.references()[0], "gender")
assert hasattr(database.references()[0], "race")
assert hasattr(database.probes()[0], "gender")
assert hasattr(database.probes()[0], "race")
p = "vgg2-short-with-eval"
database = VGG2Database(protocol=p)
......@@ -765,3 +789,8 @@ def test_vgg2():
assert len(database.references(group="eval")) == 250
assert len(database.probes(group="eval")) == 1250
assert hasattr(database.references()[0], "gender")
assert hasattr(database.references()[0], "race")
assert hasattr(database.probes()[0], "gender")
assert hasattr(database.probes()[0], "race")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment