def test_facerecognizer_save():
images1 = [
os.path.join(DATA_DIR, "sampleimages/ff1.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff2.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff3.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff4.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff5.jpg")
]
images2 = [
os.path.join(DATA_DIR, "sampleimages/fm1.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm2.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm3.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm4.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm5.jpg")
]
imgset1 = []
imgset2 = []
for img in images1:
imgset1.append(Factory.Image(img))
label1 = ["female"] * len(imgset1)
for img in images2:
imgset2.append(Factory.Image(img))
label2 = ["male"] * len(imgset2)
imgset = imgset1 + imgset2
labels = label1 + label2
f = FaceRecognizer()
trained = f.train(imgset, labels)
filename = os.path.join(tempfile.gettempdir(), "gendertrain.xml")
if (trained):
saved = f.save(filename)
assert_equal(saved, True)
if not os.path.exists(os.path.abspath(filename)):
assert False
os.remove(filename)
def test_facerecognizer_save():
images1 = [os.path.join(DATA_DIR, "sampleimages/ff1.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff2.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff3.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff4.jpg"),
os.path.join(DATA_DIR, "sampleimages/ff5.jpg")]
images2 = [os.path.join(DATA_DIR, "sampleimages/fm1.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm2.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm3.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm4.jpg"),
os.path.join(DATA_DIR, "sampleimages/fm5.jpg")]
imgset1 = []
imgset2 = []
for img in images1:
imgset1.append(Factory.Image(img))
label1 = ["female"] * len(imgset1)
for img in images2:
imgset2.append(Factory.Image(img))
label2 = ["male"] * len(imgset2)
imgset = imgset1 + imgset2
labels = label1 + label2
f = FaceRecognizer()
trained = f.train(imgset, labels)
filename = os.path.join(tempfile.gettempdir(), "gendertrain.xml")
if (trained):
saved = f.save(filename)
assert_equal(saved, True)
if not os.path.exists(os.path.abspath(filename)):
assert False
os.remove(filename)
