19 lines
514 B
Python
19 lines
514 B
Python
import skimage.data as data
|
|
from skimage.feature import Cascade
|
|
|
|
|
|
def test_detector_astronaut():
|
|
# Load the trained file from the module root.
|
|
trained_file = data.lbp_frontal_face_cascade_filename()
|
|
|
|
# Initialize the detector cascade.
|
|
detector = Cascade(trained_file)
|
|
|
|
img = data.astronaut()
|
|
|
|
detected = detector.detect_multi_scale(
|
|
img=img, scale_factor=1.2, step_ratio=1, min_size=(60, 60), max_size=(123, 123)
|
|
)
|
|
|
|
assert len(detected) == 1, 'One face should be detected.'
|