mirror of https://github.com/vladmandic/human
29 lines
601 B
JavaScript
29 lines
601 B
JavaScript
class ModelWeights {
|
|
constructor(variables) {
|
|
this.variables = variables;
|
|
}
|
|
|
|
weights(layerName) {
|
|
return this.variables[`MobilenetV1/${layerName}/weights`];
|
|
}
|
|
|
|
depthwiseBias(layerName) {
|
|
return this.variables[`MobilenetV1/${layerName}/biases`];
|
|
}
|
|
|
|
convBias(layerName) {
|
|
return this.depthwiseBias(layerName);
|
|
}
|
|
|
|
depthwiseWeights(layerName) {
|
|
return this.variables[`MobilenetV1/${layerName}/depthwise_weights`];
|
|
}
|
|
|
|
dispose() {
|
|
for (const varName in this.variables) {
|
|
this.variables[varName].dispose();
|
|
}
|
|
}
|
|
}
|
|
exports.ModelWeights = ModelWeights;
|