Skip to content

Commit 4b69fc2

Browse files
justadudewhohacksjustadudewhohacks
committed
check in latest build
1 parent b7def9f commit 4b69fc2

22 files changed

+400
-147
lines changed

‎build/commons/NeuralNetwork.d.ts

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
import * as tf from '@tensorflow/tfjs-core';
2+
import { ParamMapping } from './types';
3+
export declare class NeuralNetwork<TNetParams> {
4+
protected _params: TNetParams | undefined;
5+
protected _paramMappings: ParamMapping[];
6+
readonly params: TNetParams | undefined;
7+
readonly paramMappings: ParamMapping[];
8+
getParamFromPath(paramPath: string): tf.Tensor;
9+
reassignParamFromPath(paramPath: string, tensor: tf.Tensor): void;
10+
getParamList(): {
11+
path: string;
12+
tensor: tf.Tensor<tf.Rank>;
13+
}[];
14+
getTrainableParams(): {
15+
path: string;
16+
tensor: tf.Tensor<tf.Rank>;
17+
}[];
18+
getFrozenParams(): {
19+
path: string;
20+
tensor: tf.Tensor<tf.Rank>;
21+
}[];
22+
variable(): void;
23+
freeze(): void;
24+
dispose(): void;
25+
private traversePropertyPath(paramPath);
26+
}

‎build/commons/NeuralNetwork.js

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

‎build/commons/NeuralNetwork.js.map

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

‎build/commons/extractConvParamsFactory.d.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
import { ConvParams, ExtractWeightsFunction } from './types';
2-
export declare function extractConvParamsFactory(extractWeights: ExtractWeightsFunction): (channelsIn: number, channelsOut: number, filterSize: number) => ConvParams;
1+
import { ConvParams, ExtractWeightsFunction, ParamMapping } from './types';
2+
export declare function extractConvParamsFactory(extractWeights: ExtractWeightsFunction, paramMappings: ParamMapping[]): (channelsIn: number, channelsOut: number, filterSize: number, mappedPrefix: string) => ConvParams;

‎build/commons/extractConvParamsFactory.js

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

‎build/commons/extractConvParamsFactory.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/commons/extractWeightEntry.d.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
export declare function extractWeightEntry(weightMap: any, path: string, paramRank: number): {
2+
path: string;
3+
tensor: any;
4+
};

‎build/commons/extractWeightEntry.js

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

‎build/commons/extractWeightEntry.js.map

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

‎build/commons/types.d.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,7 @@ export declare type BatchReshapeInfo = {
1010
paddingX: number;
1111
paddingY: number;
1212
};
13+
export declare type ParamMapping = {
14+
originalPath?: string;
15+
paramPath: string;
16+
};

‎build/faceLandmarkNet/FaceLandmarkNet.d.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import * as tf from '@tensorflow/tfjs-core';
2+
import { NeuralNetwork } from '../commons/NeuralNetwork';
23
import { NetInput } from '../NetInput';
34
import { TNetInput } from '../types';
45
import { FaceLandmarks } from './FaceLandmarks';
5-
export declare class FaceLandmarkNet {
6-
private _params;
6+
import { NetParams } from './types';
7+
export declare class FaceLandmarkNet extends NeuralNetwork<NetParams> {
78
load(weightsOrUrl: Float32Array | string | undefined): Promise<void>;
89
extractWeights(weights: Float32Array): void;
910
forwardInput(input: NetInput): tf.Tensor2D;

‎build/faceLandmarkNet/FaceLandmarkNet.js

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

0 commit comments

Comments
 (0)