From 0267811fa474b191977be77846f08bb2c523483c Mon Sep 17 00:00:00 2001 From: Laurent COLBOIS <lcolbois@.idiap.ch> Date: Thu, 3 Dec 2020 14:00:34 +0100 Subject: [PATCH] Added comparison test between TF1 and TF2 for FaceNet embeddings --- .../data/facenet_sandberg_20170512-110547.hdf5 | Bin 0 -> 2560 bytes bob/bio/face/test/test_embeddings.py | 13 ++++++++++--- 2 files changed, 10 insertions(+), 3 deletions(-) create mode 100644 bob/bio/face/test/data/facenet_sandberg_20170512-110547.hdf5 diff --git a/bob/bio/face/test/data/facenet_sandberg_20170512-110547.hdf5 b/bob/bio/face/test/data/facenet_sandberg_20170512-110547.hdf5 new file mode 100644 index 0000000000000000000000000000000000000000..e70f7bde7711b507423fda171bfcf9ca8a57dbdd GIT binary patch literal 2560 zcmeD5aB<`1lHy_j0S*oZ76t(j3y%Lo0Ruz@1C;w8rXT^zV1&{gP<jOugu%!Ff(#52 z5H5ot)O>VxAwjOL3=E8L^U>`wV1XFG0ij085f%a-u8sj9FGqj~1_o#<VqiD{O=nPH zhQy+x#7YKGN`i=ZKq#_FMg}$paM}hFAagaqW-&5B1Rx{>Ljzb8BoAT0_{<CvVEuv$ z@(hq9tiT}7!604_QO3Z^zyX$LVuBdR0nI7Qj0~`(j^aZhs4&RhjFTtg85lC4Ceeoq z_VafE<!G4W6qxB_@2D-KAut*O)DM9@ib8vX!<qN_E@<1Mzfoo1-V+@*H^LU#h<|9= zTX%$OugGl|+ctr>HrCmKdt&dcv`ywWw<|i{ve)Tt-5$-pskW!8=Gw*zS?xK(KHWCp zXY1a3vOjk}RNT7L=i=JE5-V@n-g^7i_Mzkay~|iX+eXa3Yny7=ZF_i=oAr|;+xGOO z{IPwm>~7m0x@0d`o4oZtMJJmzbN1{N5Hz=qaj&-7-t*XY*D^y}Wv2w&3)1IopLNRF z89imNeLv^J-mm-4?)_)@c(3={VB2~vUR!<RqJ7&%x@^<a`}S^q`gKp!0@giJ4<_2q zuXnZ;{;#w*d9CukKSuKV%=h2jJE7TY&%M`6_xA0&vuAq3hrN5cqU@gjlDA!u`QK*y z{V1EK9vb`lW_9cdu$pE2Jo<~Zi@e_62YzX`i$DL`>%UH9A4ieMKD&%9d-L=hZ1u8v zZSPN#w3T*Aw7zwCiS55$J-bt)rM5|m8a8>6OxEja-|n@ka<|K$7iqgQoq6xIMGN-) zli;>Ex82P4Ymu3q$r-VIGrt_$vm?Q3pUuW)dj;1DS#Jo~Zj;uTZ+q#Z|Gsmx*!SJw zYTWy)XTjcgV)u4`-6CZBKV{wCcjuqmeh*${!*g%DZSAs?HmXX!HuAN)ww_<s+3q{| v)@E*5sIA_-ls#Nmx$KVnsoR#yirI-Z9^b2^UTM2UW$E6if(F~=9qu*&cDgKX literal 0 HcmV?d00001 diff --git a/bob/bio/face/test/test_embeddings.py b/bob/bio/face/test/test_embeddings.py index e97e3b00..b8bd3f0f 100644 --- a/bob/bio/face/test/test_embeddings.py +++ b/bob/bio/face/test/test_embeddings.py @@ -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 -- GitLab