diff --git a/bob/bio/face/test/test_baselines.py b/bob/bio/face/test/test_baselines.py index 4bff3b8ed6880986f49d6ad79b7046a42c06906c..b1177216d7e614b01df203909466b9078d01ad42 100644 --- a/bob/bio/face/test/test_baselines.py +++ b/bob/bio/face/test/test_baselines.py @@ -12,6 +12,7 @@ import os import bob.io.base import functools import copy +from nose.plugins.attrib import attr from bob.bio.base.test.utils import is_library_available @@ -106,31 +107,37 @@ def run_baseline(baseline, samples_for_training=[]): assert "scores" in dirs +@attr('slow') @is_library_available("tensorflow") def test_facenet_baseline(): run_baseline("facenet-sanderberg") +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv2_msceleb(): run_baseline("inception-resnetv2-msceleb") +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv2_casiawebface(): run_baseline("inception-resnetv2-casiawebface") +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv1_msceleb(): run_baseline("inception-resnetv1-msceleb") +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv1_casiawebface(): run_baseline("inception-resnetv1-casiawebface") +@attr('slow') @is_library_available("mxnet") def test_arcface_insightface(): run_baseline("arcface-insightface") diff --git a/bob/bio/face/test/test_embeddings.py b/bob/bio/face/test/test_embeddings.py index ca88c315d4d8abf8045d230a921e04bd75c27029..10b5cf27e548ba960165477eaf3708ca1716661e 100644 --- a/bob/bio/face/test/test_embeddings.py +++ b/bob/bio/face/test/test_embeddings.py @@ -4,8 +4,10 @@ import numpy as np from bob.pipelines import Sample, wrap import pkg_resources from bob.bio.base.test.utils import is_library_available +from nose.plugins.attrib import attr +@attr('slow') @is_library_available("tensorflow") def test_idiap_inceptionv2_msceleb(): from bob.bio.face.embeddings.tf2_inception_resnet import ( @@ -32,6 +34,7 @@ def test_idiap_inceptionv2_msceleb(): assert output.size == 128, output.shape +@attr('slow') @is_library_available("tensorflow") def test_idiap_inceptionv2_msceleb_memory_demanding(): from bob.bio.face.embeddings.tf2_inception_resnet import ( @@ -59,6 +62,7 @@ def test_idiap_inceptionv2_msceleb_memory_demanding(): assert output.size == 128, output.shape +@attr('slow') @is_library_available("tensorflow") def test_idiap_inceptionv2_casia(): from bob.bio.face.embeddings.tf2_inception_resnet import ( @@ -85,6 +89,7 @@ def test_idiap_inceptionv2_casia(): assert output.size == 128, output.shape +@attr('slow') @is_library_available("tensorflow") def test_idiap_inceptionv1_msceleb(): from bob.bio.face.embeddings.tf2_inception_resnet import ( @@ -111,6 +116,7 @@ def test_idiap_inceptionv1_msceleb(): assert output.size == 128, output.shape +@attr('slow') @is_library_available("tensorflow") def test_idiap_inceptionv1_casia(): from bob.bio.face.embeddings.tf2_inception_resnet import ( @@ -137,6 +143,7 @@ def test_idiap_inceptionv1_casia(): assert output.size == 128, output.shape +@attr('slow') @is_library_available("tensorflow") def test_facenet_sanderberg(): from bob.bio.face.embeddings.tf2_inception_resnet import ( @@ -163,6 +170,7 @@ def test_facenet_sanderberg(): assert output.size == 128, output.shape +@attr('slow') @is_library_available("mxnet") def test_arcface_insight_face(): from bob.bio.face.embeddings.mxnet_models import ArcFaceInsightFace diff --git a/bob/bio/face/test/test_transformers.py b/bob/bio/face/test/test_transformers.py index b0506b275b5e1f0fb7aa08672498a324e65bca38..309f344c094395f3b7dc649e1a85a6e48498fc77 100644 --- a/bob/bio/face/test/test_transformers.py +++ b/bob/bio/face/test/test_transformers.py @@ -4,6 +4,7 @@ import numpy as np from bob.pipelines import Sample, SampleSet from bob.bio.base import load_resource from bob.bio.base.test.utils import is_library_available +from nose.plugins.attrib import attr def get_fake_sample(face_size=(160, 160), eyes={"leye": (46, 107), "reye": (46, 53)}): @@ -13,6 +14,7 @@ def get_fake_sample(face_size=(160, 160), eyes={"leye": (46, 107), "reye": (46, return Sample(data, key="1", annotations=annotations) +@attr('slow') @is_library_available("tensorflow") def test_facenet_sanderberg(): transformer = load_resource("facenet-sanderberg", "transformer") @@ -24,6 +26,7 @@ def test_facenet_sanderberg(): assert transformed_sample.data.size == 128 +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv2_msceleb(): transformer = load_resource("inception-resnetv2-msceleb", "transformer") @@ -35,6 +38,7 @@ def test_inception_resnetv2_msceleb(): assert transformed_sample.data.size == 128 +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv2_casiawebface(): transformer = load_resource("inception-resnetv2-casiawebface", "transformer") @@ -46,6 +50,7 @@ def test_inception_resnetv2_casiawebface(): assert transformed_sample.data.size == 128 +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv1_msceleb(): transformer = load_resource("inception-resnetv1-msceleb", "transformer") @@ -57,6 +62,7 @@ def test_inception_resnetv1_msceleb(): assert transformed_sample.data.size == 128 +@attr('slow') @is_library_available("tensorflow") def test_inception_resnetv1_casiawebface(): transformer = load_resource("inception-resnetv1-casiawebface", "transformer")