Commit b94260e4 by vincent

fixed uncompressed model weight routes in end to end tests

parent 482becb0
...@@ -2,7 +2,7 @@ const dataFiles = [ ...@@ -2,7 +2,7 @@ const dataFiles = [
'test/images/*.jpg', 'test/images/*.jpg',
'test/images/*.png', 'test/images/*.png',
'test/data/*.json', 'test/data/*.json',
'weights/*.weights' 'weights/**/*'
].map(pattern => ({ ].map(pattern => ({
pattern, pattern,
watched: false, watched: false,
......
...@@ -9,7 +9,7 @@ describe('faceLandmarkNet', () => { ...@@ -9,7 +9,7 @@ describe('faceLandmarkNet', () => {
let faceLandmarkNet: any, imgEl: HTMLImageElement, faceLandmarkPositions: Point[] let faceLandmarkNet: any, imgEl: HTMLImageElement, faceLandmarkPositions: Point[]
beforeAll(async () => { beforeAll(async () => {
const res = await axios.get('base/weights/face_landmark_68_model.weights', { responseType: 'arraybuffer' }) const res = await axios.get('base/weights/uncompressed/face_landmark_68_model.weights', { responseType: 'arraybuffer' })
const weights = new Float32Array(res.data) const weights = new Float32Array(res.data)
faceLandmarkNet = faceapi.faceLandmarkNet(weights) faceLandmarkNet = faceapi.faceLandmarkNet(weights)
......
...@@ -7,7 +7,7 @@ describe('faceRecognitionNet', () => { ...@@ -7,7 +7,7 @@ describe('faceRecognitionNet', () => {
let faceRecognitionNet: any, imgEl: HTMLImageElement, faceDescriptor: number[] let faceRecognitionNet: any, imgEl: HTMLImageElement, faceDescriptor: number[]
beforeAll(async () => { beforeAll(async () => {
const res = await axios.get('base/weights/face_recognition_model.weights', { responseType: 'arraybuffer' }) const res = await axios.get('base/weights/uncompressed/face_recognition_model.weights', { responseType: 'arraybuffer' })
const weights = new Float32Array(res.data) const weights = new Float32Array(res.data)
faceRecognitionNet = faceapi.faceRecognitionNet(weights) faceRecognitionNet = faceapi.faceRecognitionNet(weights)
......
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