Skip to content
Snippets Groups Projects

Cross validation

Merged Anjith GEORGE requested to merge cross_validation into master
1 unresolved thread
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -394,7 +394,7 @@ def test_extractors():
@@ -394,7 +394,7 @@ def test_extractors():
from ..extractor.image import FASNetExtractor
from ..extractor.image import FASNetExtractor
extractor = FASNetExtractor()
extractor = FASNetExtractor()
# this architecture expects RGB images of size 3x224x224 channel images
# 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)
output = extractor(data)
assert output.shape[0] == 1
assert output.shape[0] == 1
Loading