diff --git a/src/constants/machineLearning/config.ts b/src/constants/machineLearning/config.ts index 718e697c5..099bf2a17 100644 --- a/src/constants/machineLearning/config.ts +++ b/src/constants/machineLearning/config.ts @@ -49,7 +49,7 @@ export const DEFAULT_ML_SYNC_CONFIG: MLSyncConfig = { minScore: 0.2, }, sceneDetection: { - method: 'Image-Scene', + method: 'ImageScene', minScore: 0.1, }, textDetection: { diff --git a/src/services/machineLearning/imageSceneService.ts b/src/services/machineLearning/imageSceneService.ts index 2b8a0ebfc..1e349ee45 100644 --- a/src/services/machineLearning/imageSceneService.ts +++ b/src/services/machineLearning/imageSceneService.ts @@ -14,7 +14,7 @@ class ImageScene implements SceneDetectionService { public constructor() { this.method = { - value: 'Image-Scene', + value: 'ImageScene', version: 1, }; } @@ -27,7 +27,7 @@ class ImageScene implements SceneDetectionService { const model = await tfjsConverter.loadGraphModel( '/models/imagescene/model.json' ); - console.log('loaded image-scene model', model, tf.getBackend()); + console.log('loaded ImageScene model', model, tf.getBackend()); this.model = model; } diff --git a/src/services/machineLearning/machineLearningFactory.ts b/src/services/machineLearning/machineLearningFactory.ts index 1b3b2d6c8..4b7c2c733 100644 --- a/src/services/machineLearning/machineLearningFactory.ts +++ b/src/services/machineLearning/machineLearningFactory.ts @@ -59,7 +59,7 @@ export class MLFactory { public static getSceneDetectionService( method: SceneDetectionMethod ): SceneDetectionService { - if (method === 'Image-Scene') { + if (method === 'ImageScene') { return imageSceneService; } diff --git a/src/types/machineLearning/index.ts b/src/types/machineLearning/index.ts index b4c57fb67..02e3721dd 100644 --- a/src/types/machineLearning/index.ts +++ b/src/types/machineLearning/index.ts @@ -95,7 +95,7 @@ export declare type FaceDetectionMethod = 'BlazeFace' | 'FaceApiSSD'; export declare type ObjectDetectionMethod = 'SSDMobileNetV2'; -export declare type SceneDetectionMethod = 'Image-Scene'; +export declare type SceneDetectionMethod = 'ImageScene'; export declare type TextDetectionMethod = 'Tesseract';