Skip to content
Snippets Groups Projects
Commit c1339770 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

[nose] Used the new decorator from bob.bio.base

parent 7cf745d4
No related branches found
No related tags found
1 merge request!79Arcface from InsightFace
Pipeline #45404 failed
......@@ -13,7 +13,7 @@ import bob.io.base
import functools
import copy
from bob.bio.base.test.utils import mxnet_available, tensorflow_available
from bob.bio.base.test.utils import is_library_available
images = dict()
images["bioref"] = (
......@@ -106,32 +106,32 @@ def run_baseline(baseline, samples_for_training=[]):
assert "scores" in dirs
@tensorflow_available
@is_library_available("tensorflow")
def test_facenet_baseline():
run_baseline("facenet-sanderberg")
@tensorflow_available
@is_library_available("tensorflow")
def test_inception_resnetv2_msceleb():
run_baseline("inception-resnetv2-msceleb")
@tensorflow_available
@is_library_available("tensorflow")
def test_inception_resnetv2_casiawebface():
run_baseline("inception-resnetv2-casiawebface")
@tensorflow_available
@is_library_available("tensorflow")
def test_inception_resnetv1_msceleb():
run_baseline("inception-resnetv1-msceleb")
@tensorflow_available
@is_library_available("tensorflow")
def test_inception_resnetv1_casiawebface():
run_baseline("inception-resnetv1-casiawebface")
@mxnet_available
@is_library_available("mxnet")
def test_arcface_insightface():
run_baseline("arcface-insightface")
......
......@@ -3,10 +3,10 @@ import bob.io.base
import numpy as np
from bob.pipelines import Sample, wrap
import pkg_resources
from bob.bio.base.test.utils import mxnet_available, tensorflow_available
from bob.bio.base.test.utils import is_library_available
@tensorflow_available
@is_library_available("tensorflow")
def test_idiap_inceptionv2_msceleb():
from bob.bio.face.embeddings import InceptionResnetv2_MsCeleb_CenterLoss_2018
......@@ -30,7 +30,7 @@ def test_idiap_inceptionv2_msceleb():
assert output.size == 128, output.shape
@tensorflow_available
@is_library_available("tensorflow")
def test_idiap_inceptionv2_casia():
from bob.bio.face.embeddings import InceptionResnetv2_Casia_CenterLoss_2018
......@@ -48,7 +48,7 @@ def test_idiap_inceptionv2_casia():
assert output.size == 128, output.shape
@tensorflow_available
@is_library_available("tensorflow")
def test_idiap_inceptionv1_msceleb():
from bob.bio.face.embeddings import InceptionResnetv1_MsCeleb_CenterLoss_2018
......@@ -66,7 +66,7 @@ def test_idiap_inceptionv1_msceleb():
assert output.size == 128, output.shape
@tensorflow_available
@is_library_available("tensorflow")
def test_idiap_inceptionv1_casia():
from bob.bio.face.embeddings import InceptionResnetv1_Casia_CenterLoss_2018
......@@ -84,7 +84,7 @@ def test_idiap_inceptionv1_casia():
assert output.size == 128, output.shape
@tensorflow_available
@is_library_available("tensorflow")
def test_facenet_sanderberg():
from bob.bio.face.embeddings import FaceNetSanderberg_20170512_110547
......@@ -101,7 +101,7 @@ def test_facenet_sanderberg():
assert output.size == 128, output.shape
@mxnet_available
@is_library_available("mxnet")
def test_arcface_insight_face():
from bob.bio.face.embeddings import ArcFaceInsightFace
......
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