Skip to content

Commit b8d6259

Browse files
justadudewhohacksjustadudewhohacks
authored
Merge pull request justadudewhohacks#33 from justadudewhohacks/facelandmarks-fix
pad face landmark net input to square and center to prevent stretching of input image
2 parents fd69881 + 4177c46 commit b8d6259

File tree

14 files changed

+256
-77
lines changed

14 files changed

+256
-77
lines changed

‎.gitignore

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,4 @@
11
node_modules
2-
.rpt2_cache
2+
.rpt2_cache
3+
.env*
4+
tmp

‎.npmignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,5 @@ node_modules
44
examples
55
test
66
proto
7-
weights
7+
weights
8+
tools

‎build/Rect.d.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,5 +10,6 @@ export declare class Rect implements IRect {
1010
width: number;
1111
height: number;
1212
constructor(x: number, y: number, width: number, height: number);
13+
toSquare(): Rect;
1314
floor(): Rect;
1415
}

‎build/Rect.js

Lines changed: 13 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎build/Rect.js.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎build/faceLandmarkNet/FaceLandmarkNet.d.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,7 @@ export declare class FaceLandmarkNet {
66
private _params;
77
load(weightsOrUrl: Float32Array | string | undefined): Promise<void>;
88
extractWeights(weights: Float32Array): void;
9+
forwardTensor(imgTensor: tf.Tensor4D): tf.Tensor2D;
10+
forward(input: tf.Tensor | NetInput | TNetInput): Promise<tf.Tensor2D>;
911
detectLandmarks(input: tf.Tensor | NetInput | TNetInput): Promise<FaceLandmarks>;
1012
}

‎build/faceLandmarkNet/FaceLandmarkNet.js

Lines changed: 71 additions & 25 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎build/faceLandmarkNet/FaceLandmarkNet.js.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)