add object detection default config

This commit is contained in:
Abhinav 2022-02-22 07:50:54 +05:30
parent 61a42245b8
commit e0f136a19e
2 changed files with 8 additions and 0 deletions

View file

@ -43,6 +43,9 @@ export const DEFAULT_ML_SYNC_CONFIG: MLSyncConfig = {
// maxDistanceInsideCluster: 0.4,
generateDebugInfo: true,
},
ObjectDetection: {
method: 'SSDMobileNetV2',
},
// tsne: {
// samples: 200,
// dim: 2,

View file

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