Skip to content
Snippets Groups Projects
Commit 60633779 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[test] fixed test for LightCNN9Extractor (input data)

parent 77e54248
No related branches found
No related tags found
No related merge requests found
Pipeline #26427 passed
......@@ -41,7 +41,7 @@ def test_lightcnn9():
"""
from . import LightCNN9Extractor
extractor = LightCNN9Extractor()
# this architecture expects 1x128x128 images
data = numpy.random.rand(1, 128, 128).astype("float32")
# this architecture expects 128x128 grayscale images
data = numpy.random.rand(128, 128).astype("float32")
output = extractor(data)
assert output.shape[0] == 256
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment