mirror of https://github.com/vladmandic/human
109 lines
6.1 KiB
JavaScript
109 lines
6.1 KiB
JavaScript
/* eslint-disable indent */
|
|
/* eslint-disable no-multi-spaces */
|
|
|
|
export default {
|
|
backend: 'webgl', // select tfjs backend to use
|
|
console: true, // enable debugging output to console
|
|
scoped: false, // enable scoped runs
|
|
// some models *may* have memory leaks, this wrapps everything in a local scope at a cost of performance
|
|
// typically not needed
|
|
videoOptimized: true, // perform additional optimizations when input is video, must be disabled for images
|
|
filter: {
|
|
enabled: true, // enable image pre-processing filters
|
|
width: 0, // resize input width
|
|
height: 0, // resize input height
|
|
// if both width and height are set to 0, there is no resizing
|
|
// if just one is set, second one is scaled automatically
|
|
// if both are set, values are used as-is
|
|
return: true, // return processed canvas imagedata in result
|
|
brightness: 0, // range: -1 (darken) to 1 (lighten)
|
|
contrast: 0, // range: -1 (reduce contrast) to 1 (increase contrast)
|
|
sharpness: 0, // range: 0 (no sharpening) to 1 (maximum sharpening)
|
|
blur: 0, // range: 0 (no blur) to N (blur radius in pixels)
|
|
saturation: 0, // range: -1 (reduce saturation) to 1 (increase saturation)
|
|
hue: 0, // range: 0 (no change) to 360 (hue rotation in degrees)
|
|
negative: false, // image negative
|
|
sepia: false, // image sepia colors
|
|
vintage: false, // image vintage colors
|
|
kodachrome: false, // image kodachrome colors
|
|
technicolor: false, // image technicolor colors
|
|
polaroid: false, // image polaroid camera effect
|
|
pixelate: 0, // range: 0 (no pixelate) to N (number of pixels to pixelate)
|
|
},
|
|
face: {
|
|
enabled: true, // controls if specified modul is enabled
|
|
// face.enabled is required for all face models: detector, mesh, iris, age, gender, emotion
|
|
// (note: module is not loaded until it is required)
|
|
detector: {
|
|
modelPath: '../models/blazeface/back/model.json', // can be 'front' or 'back'.
|
|
// 'front' is optimized for large faces such as front-facing camera and 'back' is optimized for distanct faces.
|
|
inputSize: 256, // fixed value: 128 for front and 256 for 'back'
|
|
maxFaces: 10, // maximum number of faces detected in the input, should be set to the minimum number for performance
|
|
skipFrames: 10, // how many frames to go without re-running the face bounding box detector, only used for video inputs
|
|
// if model is running st 25 FPS, we can re-use existing bounding box for updated face mesh analysis
|
|
// as face probably hasn't moved much in short time (10 * 1/25 = 0.25 sec)
|
|
minConfidence: 0.5, // threshold for discarding a prediction
|
|
iouThreshold: 0.3, // threshold for deciding whether boxes overlap too much in non-maximum suppression
|
|
scoreThreshold: 0.7, // threshold for deciding when to remove boxes based on score in non-maximum suppression
|
|
},
|
|
mesh: {
|
|
enabled: true,
|
|
modelPath: '../models/facemesh/model.json',
|
|
inputSize: 192, // fixed value
|
|
},
|
|
iris: {
|
|
enabled: true,
|
|
modelPath: '../models/iris/model.json',
|
|
enlargeFactor: 2.3, // empiric tuning
|
|
inputSize: 64, // fixed value
|
|
},
|
|
age: {
|
|
enabled: true,
|
|
modelPath: '../models/ssrnet-age/imdb/model.json', // can be 'imdb' or 'wiki'
|
|
// which determines training set for model
|
|
inputSize: 64, // fixed value
|
|
skipFrames: 10, // how many frames to go without re-running the detector, only used for video inputs
|
|
},
|
|
gender: {
|
|
enabled: true,
|
|
minConfidence: 0.8, // threshold for discarding a prediction
|
|
modelPath: '../models/ssrnet-gender/imdb/model.json',
|
|
},
|
|
emotion: {
|
|
enabled: true,
|
|
inputSize: 64, // fixed value
|
|
minConfidence: 0.5, // threshold for discarding a prediction
|
|
skipFrames: 10, // how many frames to go without re-running the detector
|
|
modelPath: '../models/emotion/model.json',
|
|
},
|
|
},
|
|
body: {
|
|
enabled: true,
|
|
modelPath: '../models/posenet/model.json',
|
|
inputResolution: 257, // fixed value
|
|
outputStride: 16, // fixed value
|
|
maxDetections: 10, // maximum number of people detected in the input, should be set to the minimum number for performance
|
|
scoreThreshold: 0.7, // threshold for deciding when to remove boxes based on score in non-maximum suppression
|
|
nmsRadius: 20, // radius for deciding points are too close in non-maximum suppression
|
|
},
|
|
hand: {
|
|
enabled: true,
|
|
inputSize: 256, // fixed value
|
|
skipFrames: 10, // how many frames to go without re-running the hand bounding box detector, only used for video inputs
|
|
// if model is running st 25 FPS, we can re-use existing bounding box for updated hand skeleton analysis
|
|
// as the hand probably hasn't moved much in short time (10 * 1/25 = 0.25 sec)
|
|
minConfidence: 0.5, // threshold for discarding a prediction
|
|
iouThreshold: 0.3, // threshold for deciding whether boxes overlap too much in non-maximum suppression
|
|
scoreThreshold: 0.7, // threshold for deciding when to remove boxes based on score in non-maximum suppression
|
|
enlargeFactor: 1.65, // empiric tuning as skeleton prediction prefers hand box with some whitespace
|
|
maxHands: 10, // maximum number of hands detected in the input, should be set to the minimum number for performance
|
|
detector: {
|
|
anchors: '../models/handdetect/anchors.json',
|
|
modelPath: '../models/handdetect/model.json',
|
|
},
|
|
skeleton: {
|
|
modelPath: '../models/handskeleton/model.json',
|
|
},
|
|
},
|
|
};
|