[mob][photos] Rename arcface

This commit is contained in:
laurenspriem
2024-06-19 17:43:22 +05:30
parent 848f3a0f6b
commit 0c790e64f8
2 changed files with 8 additions and 8 deletions

View File

@@ -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 = [
<double>[38.2946, 51.6963],
<double>[73.5318, 51.5014],
<double>[56.0252, 71.7366],
<double>[56.1396, 92.2848],
];
final arcface5Landmarks = [
final mobilefacenetIdeal5Landmarks = [
<double>[38.2946, 51.6963],
<double>[73.5318, 51.5014],
<double>[56.0252, 71.7366],
<double>[41.5493, 92.3655],
<double>[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<List<double>> src) {
_cleanParams();
final (params, center, size, rotation) =
_umeyama(src, arcfaceNormalized5, true);
_umeyama(src, mobilefacenetIdealNormalized5, true);
_params = params;
_center = center;
_size = size;

View File

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