diff --git a/src/constants/machineLearning/config.ts b/src/constants/machineLearning/config.ts index 7a05f48e0..9d25755bf 100644 --- a/src/constants/machineLearning/config.ts +++ b/src/constants/machineLearning/config.ts @@ -43,6 +43,9 @@ export const DEFAULT_ML_SYNC_CONFIG: MLSyncConfig = { // maxDistanceInsideCluster: 0.4, generateDebugInfo: true, }, + ObjectDetection: { + method: 'SSDMobileNetV2', + }, // tsne: { // samples: 200, // dim: 2, diff --git a/src/types/machineLearning/index.ts b/src/types/machineLearning/index.ts index a92c07535..474f09472 100644 --- a/src/types/machineLearning/index.ts +++ b/src/types/machineLearning/index.ts @@ -221,6 +221,10 @@ export interface FaceDetectionConfig { minFaceSize: number; } +export interface ObjectDetectionConfig { + method: ObjectDetectionMethod; +} + export interface FaceCropConfig { enabled: boolean; method: FaceCropMethod; @@ -264,6 +268,7 @@ export interface MLSyncConfig extends Config { faceAlignment: FaceAlignmentConfig; faceEmbedding: FaceEmbeddingConfig; faceClustering: FaceClusteringConfig; + ObjectDetection: ObjectDetectionConfig; tsne?: TSNEConfig; mlVersion: number; }