def test_image_recognize_face():
img = Factory.Image("lenna")
recognizer = FaceRecognizer()
faces = img.find_haar_features("face.xml")
face = faces[0].crop()
recognizer.load(os.path.join(DATA_DIR, "FaceRecognizer/GenderData.xml"))
assert_equals(face.recognize_face(recognizer)[0], 0)
# invalid recognizer
assert_is_none(img.recognize_face(2))
def test_image_recognize_face():
img = Factory.Image("lenna")
recognizer = FaceRecognizer()
faces = img.find_haar_features("face.xml")
face = faces[0].crop()
recognizer.load(os.path.join(DATA_DIR, "FaceRecognizer/GenderData.xml"))
assert_equals(face.recognize_face(recognizer)[0], 0)
# invalid recognizer
assert_is_none(img.recognize_face(2))
def test_facerecognizer_load():
f = FaceRecognizer()
trained = f.train(csvfile=os.path.join(
DATA_DIR, "test/standard/test_facerecognizer_train_data.csv"),
delimiter=",")
images3 = [
os.path.join(DATA_DIR, "sampleimages/ff1.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff5.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm3.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm4.jpg")
]
imgset3 = []
for img in images3:
imgset3.append(Factory.Image(img))
label = []
for img in imgset3:
name, confidence = f.predict(img)
label.append(name)
assert_list_equal(['female', 'female', 'male', 'male'], label)
fr1 = FaceRecognizer()
trained = fr1.load("no_such_file.xml")
assert_equal(trained, False)
prediction = fr1.predict(imgset3[0])
assert_equal(prediction, None)
def test_facerecognizer_load():
f = FaceRecognizer()
trained = f.train(
csvfile=os.path.join(DATA_DIR, "test/standard/test_facerecognizer_train_data.csv"),
delimiter=",")
images3 = [os.path.join(DATA_DIR, "sampleimages/ff1.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff5.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm3.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm4.jpg")]
imgset3 = []
for img in images3:
imgset3.append(Factory.Image(img))
label = []
for img in imgset3:
name, confidence = f.predict(img)
label.append(name)
assert_list_equal(['female', 'female', 'male', 'male'], label)
fr1 = FaceRecognizer()
trained = fr1.load("no_such_file.xml")
assert_equal(trained, False)
prediction = fr1.predict(imgset3[0])
assert_equal(prediction, None)
def test_image_find_and_recognize_faces():
img = Factory.Image("lenna")
recognizer = FaceRecognizer()
recognizer.load(os.path.join(DATA_DIR, "FaceRecognizer/GenderData.xml"))
assert_equals(
img.find_and_recognize_faces(recognizer, "face.xml")[0][1], 0)
def test_image_find_and_recognize_faces():
img = Factory.Image("lenna")
recognizer = FaceRecognizer()
recognizer.load(os.path.join(DATA_DIR, "FaceRecognizer/GenderData.xml"))
assert_equals(img.find_and_recognize_faces(recognizer, "face.xml")[0][1],
0)
