move constants

This commit is contained in:
Abhinav 2022-04-12 16:37:06 +05:30
parent 2b2b82155c
commit 080fcab6a3
6 changed files with 32 additions and 26 deletions

View file

@ -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;

View file

@ -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>;

View file

@ -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';

View file

@ -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 {

View file

@ -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>;

View file

@ -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,