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
No related branches found
No related tags found
1 merge request!84Test Facenet
Pipeline #46290 passed
File added
......@@ -93,19 +93,26 @@ def test_idiap_inceptionv1_casia():
@is_library_available("tensorflow")
def test_facenet_sanderberg():
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)
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]
assert output.size == 128, output.shape
# Sample Batch
sample = Sample(data)
transformer_sample = wrap(["sample"], transformer)
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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment