Skip to content
Snippets Groups Projects
Commit 0267811f authored by Laurent COLBOIS's avatar Laurent COLBOIS
Browse files

Added comparison test between TF1 and TF2 for FaceNet embeddings

parent d4fbeef1
Branches
Tags
1 merge request!84Test Facenet
Pipeline #46290 passed
File added
...@@ -93,19 +93,26 @@ def test_idiap_inceptionv1_casia(): ...@@ -93,19 +93,26 @@ def test_idiap_inceptionv1_casia():
@is_library_available("tensorflow") @is_library_available("tensorflow")
def test_facenet_sanderberg(): def test_facenet_sanderberg():
from bob.bio.face.embeddings.tf2_inception_resnet import ( from bob.bio.face.embeddings.tf2_inception_resnet import (
FaceNetSanderberg_20170512_110547, FaceNetSanderberg_20170512_110547
) )
reference = bob.io.base.load(
pkg_resources.resource_filename(
"bob.bio.face.test", "data/facenet_sandberg_20170512-110547.hdf5"
)
)
np.random.seed(10) np.random.seed(10)
transformer = FaceNetSanderberg_20170512_110547() transformer = FaceNetSanderberg_20170512_110547()
data = np.random.rand(3, 160, 160).astype("uint8") data = (np.random.rand(3, 160, 160) * 255).astype("uint8")
output = transformer.transform([data])[0] output = transformer.transform([data])[0]
assert output.size == 128, output.shape assert output.size == 128, output.shape
# Sample Batch # Sample Batch
sample = Sample(data) sample = Sample(data)
transformer_sample = wrap(["sample"], transformer) transformer_sample = wrap(["sample"], transformer)
output = [s.data for s in transformer_sample.transform([sample])][0] output = [s.data for s in transformer_sample.transform([sample])][0]
np.testing.assert_allclose(output, reference.flatten(), rtol=1e-5, atol=1e-4)
assert output.size == 128, output.shape assert output.size == 128, output.shape
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment