Skip to content

F
face-api.js
Commit 52858047 authored by vincent's avatar vincent
Browse files
Options

use PredictAgeAndGenderTask in ageAndGenderRecognition example + align face in... 

use PredictAgeAndGenderTask in ageAndGenderRecognition example + align face in faceExpressionRecognition example
parent 7c7cf2a8
Hide whitespace changes
Inline Side-by-side
Showing with 30 additions and 15 deletions
examples/examples-browser/views/ageAndGenderRecognition.html
View file @ 52858047
......@@ -3,7 +3,6 @@
<head>
<script src="face-api.js"></script>
<script src="js/commons.js"></script>
<script src="js/drawing.js"></script>
<script src="js/faceDetectionControls.js"></script>
<script src="js/imageSelectionControls.js"></script>
<link rel="stylesheet" href="styles.css">
......@@ -139,8 +138,6 @@
<script>
window.net = new faceapi.AgeGenderNet()
async function updateResults() {
if (!isFaceDetectionModelLoaded()) {
return
......@@ -149,24 +146,37 @@
const inputImgEl = $('#inputImg').get(0)
const options = getFaceDetectorOptions()
const result = await faceapi.detectSingleFace(inputImgEl, options)
if (!result) return
const face = (await faceapi.extractFaces(inputImgEl, [result]))[0]
const { age, gender, genderProbability } = await window.net.predictAgeAndGender(face)
console.log('age', age)
console.log('gender', gender, genderProbability)
const results = await faceapi.detectAllFaces(inputImgEl, options)
// compute face landmarks to align faces for better accuracy
.withFaceLandmarks()
.withAgeAndGender()
const canvas = $('#overlay').get(0)
faceapi.matchDimensions(canvas, inputImgEl)
const resizedResults = faceapi.resizeResults(results, inputImgEl)
faceapi.draw.drawDetections(canvas, resizedResults)
resizedResults.forEach(result => {
const { age, gender, genderProbability } = result
new faceapi.draw.DrawTextField(
[
`${faceapi.round(age, 0)} years`,
`${gender} (${faceapi.round(genderProbability)})`
],
result.detection.box.bottomLeft
).draw(canvas)
})
}
async function run() {
// load face detection and face expression recognition models
await changeFaceDetector(SSD_MOBILENETV1)
await faceapi.loadFaceLandmarkModel('/')
// TODO
const weights = await faceapi.fetchNetWeights('tmp/age_gender.weights')
console.log(weights.length)
await window.net.load(weights)
await faceapi.nets.ageGenderNet.load(weights)
// start processing image
updateResults()
......
examples/examples-browser/views/faceExpressionRecognition.html
View file @ 52858047
......@@ -148,7 +148,10 @@
const inputImgEl = $('#inputImg').get(0)
const options = getFaceDetectorOptions()
const results = await faceapi.detectAllFaces(inputImgEl, options).withFaceExpressions()
const results = await faceapi.detectAllFaces(inputImgEl, options)
// compute face landmarks to align faces for better accuracy
.withFaceLandmarks()
.withFaceExpressions()
const canvas = $('#overlay').get(0)
faceapi.matchDimensions(canvas, inputImgEl)
......@@ -161,7 +164,9 @@
async function run() {
// load face detection and face expression recognition models
// and load face landmark model for face alignment
await changeFaceDetector(SSD_MOBILENETV1)
await faceapi.loadFaceLandmarkModel('/')
await faceapi.loadFaceExpressionModel('/')
// start processing image
updateResults()
......
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