fixed uncompressed model weight routes in end to end tests

justadudewhohacks · justadudewhohacks · commit b94260e47018 · 2018-06-21T11:19:29.000+02:00
diff --git a/karma.conf.js b/karma.conf.js
@@ -2,7 +2,7 @@ const dataFiles = [
   'test/images/*.jpg',
   'test/images/*.png',
   'test/data/*.json',
-  'weights/*.weights'
+  'weights/**/*'
 ].map(pattern => ({
   pattern,
   watched: false,
diff --git a/test/e2e/faceLandmarkNet.test.ts b/test/e2e/faceLandmarkNet.test.ts
@@ -9,7 +9,7 @@ describe('faceLandmarkNet', () => {
   let faceLandmarkNet: any, imgEl: HTMLImageElement, faceLandmarkPositions: Point[]
 
   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)
     faceLandmarkNet = faceapi.faceLandmarkNet(weights)
 
diff --git a/test/e2e/faceRecognitionNet.test.ts b/test/e2e/faceRecognitionNet.test.ts
@@ -7,7 +7,7 @@ describe('faceRecognitionNet', () => {
   let faceRecognitionNet: any, imgEl: HTMLImageElement, faceDescriptor: number[]
 
   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)
     faceRecognitionNet = faceapi.faceRecognitionNet(weights)
 
