From 0c790e64f8bada3c4f604313d89633de916d6773 Mon Sep 17 00:00:00 2001 From: laurenspriem Date: Wed, 19 Jun 2024 17:43:22 +0530 Subject: [PATCH] [mob][photos] Rename arcface --- .../face_ml/face_alignment/similarity_transform.dart | 10 +++++----- .../machine_learning/face_ml/face_ml_methods.dart | 6 +++--- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/mobile/lib/services/machine_learning/face_ml/face_alignment/similarity_transform.dart b/mobile/lib/services/machine_learning/face_ml/face_alignment/similarity_transform.dart index 0d8e7ab3ae..8d995342ba 100644 --- a/mobile/lib/services/machine_learning/face_ml/face_alignment/similarity_transform.dart +++ b/mobile/lib/services/machine_learning/face_ml/face_alignment/similarity_transform.dart @@ -17,23 +17,23 @@ class SimilarityTransform { double _size = 1; // 1 / scale double _rotation = 0; // atan2(simRotation[1][0], simRotation[0][0]); - final arcface4Landmarks = [ + final mobilefacenetIdeal4Landmarks = [ [38.2946, 51.6963], [73.5318, 51.5014], [56.0252, 71.7366], [56.1396, 92.2848], ]; - final arcface5Landmarks = [ + final mobilefacenetIdeal5Landmarks = [ [38.2946, 51.6963], [73.5318, 51.5014], [56.0252, 71.7366], [41.5493, 92.3655], [70.7299, 92.2041], ]; - get arcfaceNormalized4 => arcface4Landmarks + get mobilefacenetIdealNormalized4 => mobilefacenetIdeal4Landmarks .map((list) => list.map((value) => value / 112.0).toList()) .toList(); - get arcfaceNormalized5 => arcface5Landmarks + get mobilefacenetIdealNormalized5 => mobilefacenetIdeal5Landmarks .map((list) => list.map((value) => value / 112.0).toList()) .toList(); @@ -69,7 +69,7 @@ class SimilarityTransform { (AlignmentResult, bool) estimate(List> src) { _cleanParams(); final (params, center, size, rotation) = - _umeyama(src, arcfaceNormalized5, true); + _umeyama(src, mobilefacenetIdealNormalized5, true); _params = params; _center = center; _size = size; diff --git a/mobile/lib/services/machine_learning/face_ml/face_ml_methods.dart b/mobile/lib/services/machine_learning/face_ml/face_ml_methods.dart index 5745234b58..70c1d03ff6 100644 --- a/mobile/lib/services/machine_learning/face_ml/face_ml_methods.dart +++ b/mobile/lib/services/machine_learning/face_ml/face_ml_methods.dart @@ -39,13 +39,13 @@ class FaceAlignmentMethod extends VersionedMethod { /// Creates a [FaceAlignmentMethod] instance with 'Empty method' as the method, and a specific `version` (default `1`) const FaceAlignmentMethod.empty() : super.empty(); - /// Creates a [FaceAlignmentMethod] instance with 'ArcFace' as the method, and a specific `version` (default `1`) - FaceAlignmentMethod.arcFace({int version = 1}) : super('ArcFace', version); + /// Creates a [FaceAlignmentMethod] instance with 'affineTransform' as the method, and a specific `version` (default `1`) + FaceAlignmentMethod.affineTransform({int version = 1}) : super('affineTransform', version); static FaceAlignmentMethod fromMlVersion(int version) { switch (version) { case 1: - return FaceAlignmentMethod.arcFace(version: version); + return FaceAlignmentMethod.affineTransform(version: version); default: return const FaceAlignmentMethod.empty(); }