Skip to content
Snippets Groups Projects
Commit 73eabc81 authored by Yannick DAYER's avatar Yannick DAYER Committed by Tiago de Freitas Pereira
Browse files

Fix transformers naming in tests

parent 176353c2
No related branches found
No related tags found
1 merge request!64Dask pipelines
......@@ -13,3 +13,4 @@ sphinx
dist
record.txt
build/
bob/bio/face/embeddings/data
......@@ -104,23 +104,23 @@ def run_baseline(baseline, samples_for_training=[]):
def test_facenet_baseline():
run_baseline("facenet_sanderberg")
run_baseline("facenet-sanderberg")
def test_inception_resnetv2_msceleb():
run_baseline("inception_resnetv2_msceleb")
run_baseline("inception-resnetv2-msceleb")
def test_inception_resnetv2_casiawebface():
run_baseline("inception_resnetv2_casiawebface")
run_baseline("inception-resnetv2-casiawebface")
def test_inception_resnetv1_msceleb():
run_baseline("inception_resnetv1_msceleb")
run_baseline("inception-resnetv1-msceleb")
def test_inception_resnetv1_casiawebface():
run_baseline("inception_resnetv1_casiawebface")
run_baseline("inception-resnetv1-casiawebface")
def test_arcface_insight_tf():
......@@ -128,7 +128,7 @@ def test_arcface_insight_tf():
tf.compat.v1.reset_default_graph()
run_baseline("arcface_insight_tf")
run_baseline("arcface-insight-tf")
def test_gabor_graph():
......
......@@ -12,7 +12,7 @@ def get_fake_sample(face_size=(160, 160), eyes={"leye": (46, 107), "reye": (46,
def test_facenet():
transformer = load_resource("facenet_sanderberg", "transformer")
transformer = load_resource("facenet-sanderberg", "transformer")
fake_sample = get_fake_sample()
......@@ -22,7 +22,7 @@ def test_facenet():
def test_inception_resnetv2_msceleb():
transformer = load_resource("inception_resnetv2_msceleb", "transformer")
transformer = load_resource("inception-resnetv2-msceleb", "transformer")
fake_sample = get_fake_sample()
......@@ -32,7 +32,7 @@ def test_inception_resnetv2_msceleb():
def test_inception_resnetv2_casiawebface():
transformer = load_resource("inception_resnetv2_casiawebface", "transformer")
transformer = load_resource("inception-resnetv2-casiawebface", "transformer")
fake_sample = get_fake_sample()
......@@ -42,7 +42,7 @@ def test_inception_resnetv2_casiawebface():
def test_inception_resnetv1_msceleb():
transformer = load_resource("inception_resnetv1_msceleb", "transformer")
transformer = load_resource("inception-resnetv1-msceleb", "transformer")
fake_sample = get_fake_sample()
......@@ -52,7 +52,7 @@ def test_inception_resnetv1_msceleb():
def test_inception_resnetv1_casiawebface():
transformer = load_resource("inception_resnetv1_casiawebface", "transformer")
transformer = load_resource("inception-resnetv1-casiawebface", "transformer")
fake_sample = get_fake_sample()
......@@ -64,7 +64,7 @@ def test_inception_resnetv1_casiawebface():
def test_arcface_insight_tf():
import tensorflow as tf
tf.compat.v1.reset_default_graph()
transformer = load_resource("arcface_insight_tf", "transformer")
transformer = load_resource("arcface-insight-tf", "transformer")
fake_sample = get_fake_sample()
......@@ -74,7 +74,7 @@ def test_arcface_insight_tf():
def test_gabor_graph():
transformer = load_resource("gabor_graph", "transformer")
transformer = load_resource("gabor-graph", "transformer")
fake_sample = get_fake_sample()
transformed_sample = transformer.transform([fake_sample])[0]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment