Remove more tf

This commit is contained in:
Manav Rathi 2024-04-11 11:15:15 +05:30
parent 3182d67ca1
commit e7902c5d83
No known key found for this signature in database

View file

@ -1,17 +1,8 @@
import * as tf from "@tensorflow/tfjs-core";
import { Matrix, inverse } from "ml-matrix";
import { Matrix } from "ml-matrix";
import { getSimilarityTransformation } from "similarity-transformation";
import { Dimensions } from "types/image";
import { FaceAlignment, FaceDetection } from "types/machineLearning";
import { cropWithRotation, transform } from "utils/image";
import {
computeRotation,
enlargeBox,
extractFaces,
getBoxCenter,
getBoxCenterPt,
toTensor4D,
} from ".";
import { Box, Point } from "../../../thirdparty/face-api/classes";
export function normalizeLandmarks(
@ -97,40 +88,6 @@ export function getArcfaceAlignment(
);
}
export function extractFaceImage(
image: tf.Tensor4D,
alignment: FaceAlignment,
faceSize: number,
) {
const affineMat = new Matrix(alignment.affineMatrix);
const I = inverse(affineMat);
return tf.tidy(() => {
const projection = tf.tensor2d([
[
I.get(0, 0),
I.get(0, 1),
I.get(0, 2),
I.get(1, 0),
I.get(1, 1),
I.get(1, 2),
0,
0,
],
]);
const faceImage = tf.image.transform(
image,
projection,
"bilinear",
"constant",
0,
[faceSize, faceSize],
);
return faceImage;
});
}
export function getAlignedFaceBox(alignment: FaceAlignment) {
return new Box({
x: alignment.center.x - alignment.size / 2,