@vladmandic/face-api
Version:
JavaScript module for Face Detection and Face Recognition Using Tensorflow/JS
40 lines (33 loc) • 1.17 kB
text/typescript
import { createFileSystem } from './createFileSystem';
import { Environment } from './types';
export function createNodejsEnv(): Environment {
const Canvas = global['Canvas'] || global['HTMLCanvasElement']
const Image = global['Image'] || global['HTMLImageElement']
const createCanvasElement = function() {
if (Canvas) {
return new Canvas()
}
throw new Error('createCanvasElement - missing Canvas implementation for nodejs environment')
}
const createImageElement = function() {
if (Image) {
return new Image()
}
throw new Error('createImageElement - missing Image implementation for nodejs environment')
}
const fetch = global['fetch'] || function() {
throw new Error('fetch - missing fetch implementation for nodejs environment')
}
const fileSystem = createFileSystem()
return {
Canvas: Canvas || class {},
CanvasRenderingContext2D: global['CanvasRenderingContext2D'] || class {},
Image: Image || class {},
ImageData: global['ImageData'] || class {},
Video: global['HTMLVideoElement'] || class {},
createCanvasElement,
createImageElement,
fetch,
...fileSystem
}
}