it('can be trained', () => { const recognizer = createFaceRecognizer() const trainFaces = loadTrainFaces() recognizer.addFaces(trainFaces.sheldon, 'sheldon') recognizer.addFaces(trainFaces.raj, 'raj') const modelState = recognizer.getDescriptorState() expect(modelState).to.be.an('array').lengthOf(2) expect(modelState).to.deep.have.members([ { className: 'sheldon', numFaces: 2 }, { className: 'raj', numFaces: 2 } ]) })
it('can be trained', (done) => { const recognizer = createFaceRecognizer() const trainFaces = loadTrainFaces() Promise.all([ recognizer.addFaces(trainFaces.sheldon, 'sheldon'), recognizer.addFaces(trainFaces.raj, 'raj') ]) .then(() => { const modelState = recognizer.getDescriptorState() expect(modelState).to.be.an('array').lengthOf(2) expect(modelState).to.deep.have.members([ { className: 'sheldon', numFaces: 2 }, { className: 'raj', numFaces: 2 } ]) done() }) .catch(err => done(err)) })
var successResponse = function(expectedBody, done) { if (!done) { done = expectedBody; expectedBody = {success: true}; } return function(err, res, body) { if (body.error) console.log(body.error); Expect(err).to.equal(null); Expect(res.statusCode).to.equal(200); Expect(body.error).to.equal(undefined); if (!expectedBody) { Expect(body.success).to.equal(true); } else { if (Array.isArray(expectedBody)) { Expect(body).to.deep.have.members(expectedBody); Expect(expectedBody).to.deep.have.members(body); } if (typeof expectedBody === 'function') { expectedBody(body); } else { Expect(body).to.deep.equal(expectedBody); } } done(); } }
it('can be trained with jittering', () => { const recognizer = createFaceRecognizer() const numJitters = 1; const trainFaces = loadTrainFaces() recognizer.addFaces(trainFaces.sheldon, 'sheldon', numJitters) recognizer.addFaces(trainFaces.raj, 'raj', numJitters) const modelState = recognizer.getDescriptorState() expect(modelState).to.be.an('array').lengthOf(2) expect(modelState).to.deep.have.members([ { className: 'sheldon', numFaces: 4 }, { className: 'raj', numFaces: 4 } ]) })
it('can be trained with jittering', (done) => { const recognizer = createFaceRecognizer() const numJitters = 1; const trainFaces = loadTrainFaces() Promise.all([ recognizer.addFaces(trainFaces.sheldon, 'sheldon', numJitters), recognizer.addFaces(trainFaces.raj, 'raj', numJitters) ]) .then(() => { const modelState = recognizer.getDescriptorState() expect(modelState).to.be.an('array').lengthOf(2) expect(modelState).to.deep.have.members([ { className: 'sheldon', numFaces: 4 }, { className: 'raj', numFaces: 4 } ]) done() }) .catch(err => done(err)) })