Commit 90453ad1 authored by Anjith GEORGE's avatar Anjith GEORGE

small fix in unit tests

parent 8ab4559f
Pipeline #27526 passed with stage
in 23 minutes and 22 seconds
......@@ -394,7 +394,7 @@ def test_extractors():
from ..extractor.image import FASNetExtractor
extractor = FASNetExtractor()
# this architecture expects RGB images of size 3x224x224 channel images
data = numpy.random.rand(3, 224, 224).astype("float32")
data = numpy.random.rand(3, 224, 224).astype("uint8")
output = extractor(data)
assert output.shape[0] == 1
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment