File size: 1,766 Bytes
bf237c2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | import * as React from 'react';
import { forwardRef, useEffect, useImperativeHandle, useContext, createContext } from 'react';
import { suspend, clear } from 'suspend-react';
/* eslint react-hooks/exhaustive-deps: 1 */
const FaceLandmarkerContext = /* @__PURE__ */createContext({});
const FaceLandmarkerDefaults = {
basePath: 'https://cdn.jsdelivr.net/npm/@mediapipe/tasks-vision@0.10.17/wasm',
options: {
baseOptions: {
modelAssetPath: 'https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/1/face_landmarker.task',
delegate: 'GPU'
},
runningMode: 'VIDEO',
outputFaceBlendshapes: true,
outputFacialTransformationMatrixes: true
}
};
const FaceLandmarker = /*#__PURE__*/forwardRef(({
basePath = FaceLandmarkerDefaults.basePath,
options = FaceLandmarkerDefaults.options,
children
}, fref) => {
const opts = JSON.stringify(options);
const faceLandmarker = suspend(async () => {
const {
FilesetResolver,
FaceLandmarker
} = await import('@mediapipe/tasks-vision');
const vision = await FilesetResolver.forVisionTasks(basePath);
return FaceLandmarker.createFromOptions(vision, options);
}, [basePath, opts]);
useEffect(() => {
return () => {
faceLandmarker == null || faceLandmarker.close();
clear([basePath, opts]);
};
}, [faceLandmarker, basePath, opts]);
useImperativeHandle(fref, () => faceLandmarker, [faceLandmarker]); // expose faceLandmarker through ref
return /*#__PURE__*/React.createElement(FaceLandmarkerContext.Provider, {
value: faceLandmarker
}, children);
});
function useFaceLandmarker() {
return useContext(FaceLandmarkerContext);
}
export { FaceLandmarker, FaceLandmarkerDefaults, useFaceLandmarker };
|