move constants
This commit is contained in:
parent
2b2b82155c
commit
080fcab6a3
|
@ -73,3 +73,15 @@ export const MAX_FACE_DISTANCE_PERCENT = Math.sqrt(2) / 100;
|
||||||
export const MAX_ML_SYNC_ERROR_COUNT = 4;
|
export const MAX_ML_SYNC_ERROR_COUNT = 4;
|
||||||
|
|
||||||
export const TEXT_DETECTION_TIMEOUT_MS = [10000, 30000, 60000, 120000, 240000];
|
export const TEXT_DETECTION_TIMEOUT_MS = [10000, 30000, 60000, 120000, 240000];
|
||||||
|
|
||||||
|
export const BLAZEFACE_MAX_FACES = 50;
|
||||||
|
export const BLAZEFACE_INPUT_SIZE = 256;
|
||||||
|
export const BLAZEFACE_IOU_THRESHOLD = 0.3;
|
||||||
|
export const BLAZEFACE_SCORE_THRESHOLD = 0.75;
|
||||||
|
export const BLAZEFACE_PASS1_SCORE_THRESHOLD = 0.4;
|
||||||
|
export const BLAZEFACE_FACE_SIZE = 112;
|
||||||
|
export const MOBILEFACENET_FACE_SIZE = 112;
|
||||||
|
|
||||||
|
export const TESSERACT_MIN_IMAGE_WIDTH = 44;
|
||||||
|
export const TESSERACT_MIN_IMAGE_HEIGHT = 20;
|
||||||
|
export const TESSERACT_MAX_IMAGE_DIMENSION = 720;
|
||||||
|
|
|
@ -6,12 +6,6 @@ import {
|
||||||
import * as tf from '@tensorflow/tfjs-core';
|
import * as tf from '@tensorflow/tfjs-core';
|
||||||
import { GraphModel } from '@tensorflow/tfjs-converter';
|
import { GraphModel } from '@tensorflow/tfjs-converter';
|
||||||
import {
|
import {
|
||||||
BLAZEFACE_FACE_SIZE,
|
|
||||||
BLAZEFACE_INPUT_SIZE,
|
|
||||||
BLAZEFACE_IOU_THRESHOLD,
|
|
||||||
BLAZEFACE_MAX_FACES,
|
|
||||||
BLAZEFACE_PASS1_SCORE_THRESHOLD,
|
|
||||||
BLAZEFACE_SCORE_THRESHOLD,
|
|
||||||
FaceDetection,
|
FaceDetection,
|
||||||
FaceDetectionMethod,
|
FaceDetectionMethod,
|
||||||
FaceDetectionService,
|
FaceDetectionService,
|
||||||
|
@ -30,7 +24,15 @@ import {
|
||||||
removeDuplicateDetections,
|
removeDuplicateDetections,
|
||||||
transformPaddedToImage,
|
transformPaddedToImage,
|
||||||
} from 'utils/machineLearning/faceDetection';
|
} from 'utils/machineLearning/faceDetection';
|
||||||
import { MAX_FACE_DISTANCE_PERCENT } from 'constants/machineLearning/config';
|
import {
|
||||||
|
BLAZEFACE_FACE_SIZE,
|
||||||
|
BLAZEFACE_INPUT_SIZE,
|
||||||
|
BLAZEFACE_IOU_THRESHOLD,
|
||||||
|
BLAZEFACE_MAX_FACES,
|
||||||
|
BLAZEFACE_PASS1_SCORE_THRESHOLD,
|
||||||
|
BLAZEFACE_SCORE_THRESHOLD,
|
||||||
|
MAX_FACE_DISTANCE_PERCENT,
|
||||||
|
} from 'constants/machineLearning/config';
|
||||||
|
|
||||||
class BlazeFaceDetectionService implements FaceDetectionService {
|
class BlazeFaceDetectionService implements FaceDetectionService {
|
||||||
private blazeFaceModel: Promise<BlazeFaceModel>;
|
private blazeFaceModel: Promise<BlazeFaceModel>;
|
||||||
|
|
|
@ -1,11 +1,11 @@
|
||||||
import * as tf from '@tensorflow/tfjs-core';
|
import * as tf from '@tensorflow/tfjs-core';
|
||||||
import { TFLiteModel } from '@tensorflow/tfjs-tflite';
|
import { TFLiteModel } from '@tensorflow/tfjs-tflite';
|
||||||
|
import { MOBILEFACENET_FACE_SIZE } from 'constants/machineLearning/config';
|
||||||
import PQueue from 'p-queue';
|
import PQueue from 'p-queue';
|
||||||
import {
|
import {
|
||||||
FaceEmbedding,
|
FaceEmbedding,
|
||||||
FaceEmbeddingMethod,
|
FaceEmbeddingMethod,
|
||||||
FaceEmbeddingService,
|
FaceEmbeddingService,
|
||||||
MOBILEFACENET_FACE_SIZE,
|
|
||||||
Versioned,
|
Versioned,
|
||||||
} from 'types/machineLearning';
|
} from 'types/machineLearning';
|
||||||
import { imageBitmapsToTensor4D } from 'utils/machineLearning';
|
import { imageBitmapsToTensor4D } from 'utils/machineLearning';
|
||||||
|
|
|
@ -1,7 +1,4 @@
|
||||||
import {
|
import {
|
||||||
TESSERACT_MAX_IMAGE_DIMENSION,
|
|
||||||
TESSERACT_MIN_IMAGE_HEIGHT,
|
|
||||||
TESSERACT_MIN_IMAGE_WIDTH,
|
|
||||||
TextDetectionMethod,
|
TextDetectionMethod,
|
||||||
TextDetectionService,
|
TextDetectionService,
|
||||||
Versioned,
|
Versioned,
|
||||||
|
@ -14,7 +11,12 @@ import { imageBitmapToBlob, resizeToSquare } from 'utils/image';
|
||||||
import { getFileType } from 'services/upload/readFileService';
|
import { getFileType } from 'services/upload/readFileService';
|
||||||
import { FILE_TYPE } from 'constants/file';
|
import { FILE_TYPE } from 'constants/file';
|
||||||
import { makeID } from 'utils/user';
|
import { makeID } from 'utils/user';
|
||||||
import { TEXT_DETECTION_TIMEOUT_MS } from 'constants/machineLearning/config';
|
import {
|
||||||
|
TESSERACT_MAX_IMAGE_DIMENSION,
|
||||||
|
TESSERACT_MIN_IMAGE_HEIGHT,
|
||||||
|
TESSERACT_MIN_IMAGE_WIDTH,
|
||||||
|
TEXT_DETECTION_TIMEOUT_MS,
|
||||||
|
} from 'constants/machineLearning/config';
|
||||||
import { promiseWithTimeout } from 'utils/common/promiseTimeout';
|
import { promiseWithTimeout } from 'utils/common/promiseTimeout';
|
||||||
|
|
||||||
class TesseractService implements TextDetectionService {
|
class TesseractService implements TextDetectionService {
|
||||||
|
|
|
@ -363,18 +363,6 @@ export interface MLLibraryData {
|
||||||
|
|
||||||
export declare type MLIndex = 'files' | 'people';
|
export declare type MLIndex = 'files' | 'people';
|
||||||
|
|
||||||
export const BLAZEFACE_MAX_FACES = 50;
|
|
||||||
export const BLAZEFACE_INPUT_SIZE = 256;
|
|
||||||
export const BLAZEFACE_IOU_THRESHOLD = 0.3;
|
|
||||||
export const BLAZEFACE_SCORE_THRESHOLD = 0.75;
|
|
||||||
export const BLAZEFACE_PASS1_SCORE_THRESHOLD = 0.4;
|
|
||||||
export const BLAZEFACE_FACE_SIZE = 112;
|
|
||||||
export const MOBILEFACENET_FACE_SIZE = 112;
|
|
||||||
|
|
||||||
export const TESSERACT_MIN_IMAGE_WIDTH = 44;
|
|
||||||
export const TESSERACT_MIN_IMAGE_HEIGHT = 20;
|
|
||||||
export const TESSERACT_MAX_IMAGE_DIMENSION = 720;
|
|
||||||
|
|
||||||
export interface FaceDetectionService {
|
export interface FaceDetectionService {
|
||||||
method: Versioned<FaceDetectionMethod>;
|
method: Versioned<FaceDetectionMethod>;
|
||||||
// init(): Promise<void>;
|
// init(): Promise<void>;
|
||||||
|
|
|
@ -1,6 +1,9 @@
|
||||||
import { NormalizedFace } from 'blazeface-back';
|
import { NormalizedFace } from 'blazeface-back';
|
||||||
import * as tf from '@tensorflow/tfjs-core';
|
import * as tf from '@tensorflow/tfjs-core';
|
||||||
import { ML_SYNC_DOWNLOAD_TIMEOUT_MS } from 'constants/machineLearning/config';
|
import {
|
||||||
|
BLAZEFACE_FACE_SIZE,
|
||||||
|
ML_SYNC_DOWNLOAD_TIMEOUT_MS,
|
||||||
|
} from 'constants/machineLearning/config';
|
||||||
import { euclidean } from 'hdbscan';
|
import { euclidean } from 'hdbscan';
|
||||||
import PQueue from 'p-queue';
|
import PQueue from 'p-queue';
|
||||||
import DownloadManager from 'services/downloadManager';
|
import DownloadManager from 'services/downloadManager';
|
||||||
|
@ -10,7 +13,6 @@ import { Dimensions } from 'types/image';
|
||||||
import {
|
import {
|
||||||
Thing,
|
Thing,
|
||||||
AlignedFace,
|
AlignedFace,
|
||||||
BLAZEFACE_FACE_SIZE,
|
|
||||||
DetectedFace,
|
DetectedFace,
|
||||||
DetectedObject,
|
DetectedObject,
|
||||||
Face,
|
Face,
|
||||||
|
|
Loading…
Reference in a new issue