From e0f136a19e66fd18e221c084e95e22f06792a5ad Mon Sep 17 00:00:00 2001
From: Abhinav <abhinavk.grd@gmail.com>
Date: Tue, 22 Feb 2022 07:50:54 +0530
Subject: [PATCH] add object detection default config

---
 src/constants/machineLearning/config.ts | 3 +++
 src/types/machineLearning/index.ts      | 5 +++++
 2 files changed, 8 insertions(+)

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