Commit b8841daa by vincent

fixed score distance calculation + increase descriptor delta

parent c66dbbde
...@@ -25,7 +25,7 @@ export function expectFullFaceDescriptions( ...@@ -25,7 +25,7 @@ export function expectFullFaceDescriptions(
expectedFullFaceDescriptions.forEach((expected, i) => { expectedFullFaceDescriptions.forEach((expected, i) => {
const { detection, landmarks, descriptor } = sortedResults[i] const { detection, landmarks, descriptor } = sortedResults[i]
expect(detection.score - expected.score).toBeLessThan(deltas.maxScoreDelta) expect(Math.abs(detection.score - expected.score)).toBeLessThan(deltas.maxScoreDelta)
expectRectClose(detection.box, expected.detection, deltas.maxBoxDelta) expectRectClose(detection.box, expected.detection, deltas.maxBoxDelta)
landmarks.positions.forEach((pt, j) => expectPointClose(pt, expected.landmarks[j], deltas.maxLandmarksDelta)) landmarks.positions.forEach((pt, j) => expectPointClose(pt, expected.landmarks[j], deltas.maxLandmarksDelta))
expect(euclideanDistance(descriptor, expected.descriptor)).toBeLessThan(deltas.maxDescriptorDelta) expect(euclideanDistance(descriptor, expected.descriptor)).toBeLessThan(deltas.maxDescriptorDelta)
......
...@@ -64,7 +64,7 @@ describe('ssdMobilenetv1', () => { ...@@ -64,7 +64,7 @@ describe('ssdMobilenetv1', () => {
maxScoreDelta: 0.01, maxScoreDelta: 0.01,
maxBoxDelta: 5, maxBoxDelta: 5,
maxLandmarksDelta: 2, maxLandmarksDelta: 2,
maxDescriptorDelta: 0.1 maxDescriptorDelta: 0.2
} }
expect(results.length).toEqual(6) expect(results.length).toEqual(6)
expectFullFaceDescriptions(results, expectedFullFaceDescriptions, expectedScores, deltas) expectFullFaceDescriptions(results, expectedFullFaceDescriptions, expectedScores, deltas)
......
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