beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix name

Browse source
This commit is contained in:
Rushikesh Tote 2022-06-01 13:14:48 +05:30
parent 090a523343
commit de3bdfecb3
4 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/constants/machineLearning/config.ts
View file

@ -49,7 +49,7 @@ export const DEFAULT_ML_SYNC_CONFIG: MLSyncConfig = {
minScore: 0.2, minScore: 0.2,
}, },
sceneDetection: { sceneDetection: {
method: 'Image-Scene', method: 'ImageScene',
minScore: 0.1, minScore: 0.1,
}, },
textDetection: { textDetection: {

4
src/services/machineLearning/imageSceneService.ts
View file

@ -14,7 +14,7 @@ class ImageScene implements SceneDetectionService {
public constructor() { public constructor() {
this.method = { this.method = {
value: 'Image-Scene', value: 'ImageScene',
version: 1, version: 1,
}; };
} }
@ -27,7 +27,7 @@ class ImageScene implements SceneDetectionService {
const model = await tfjsConverter.loadGraphModel( const model = await tfjsConverter.loadGraphModel(
'/models/imagescene/model.json' '/models/imagescene/model.json'
); );
console.log('loaded image-scene model', model, tf.getBackend()); console.log('loaded ImageScene model', model, tf.getBackend());
this.model = model; this.model = model;
} }

2
src/services/machineLearning/machineLearningFactory.ts
View file

@ -59,7 +59,7 @@ export class MLFactory {
public static getSceneDetectionService( public static getSceneDetectionService(
method: SceneDetectionMethod method: SceneDetectionMethod
): SceneDetectionService { ): SceneDetectionService {
if (method === 'Image-Scene') { if (method === 'ImageScene') {
return imageSceneService; return imageSceneService;
} }

2
src/types/machineLearning/index.ts
View file

@ -95,7 +95,7 @@ export declare type FaceDetectionMethod = 'BlazeFace' | 'FaceApiSSD';
export declare type ObjectDetectionMethod = 'SSDMobileNetV2'; export declare type ObjectDetectionMethod = 'SSDMobileNetV2';
export declare type SceneDetectionMethod = 'Image-Scene'; export declare type SceneDetectionMethod = 'ImageScene';
export declare type TextDetectionMethod = 'Tesseract'; export declare type TextDetectionMethod = 'Tesseract';