@tensorflow-models/body-pix
Version:
Pretrained BodyPix model in TensorFlow.js
27 lines • 1.14 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
var RESNET50_BASE_URL = 'https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/';
var MOBILENET_BASE_URL = 'https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/mobilenet/';
function resNet50SavedModel(stride, quantBytes) {
var graphJson = "model-stride" + stride + ".json";
if (quantBytes === 4) {
return RESNET50_BASE_URL + "float/" + graphJson;
}
else {
return RESNET50_BASE_URL + ("quant" + quantBytes + "/") + graphJson;
}
}
exports.resNet50SavedModel = resNet50SavedModel;
function mobileNetSavedModel(stride, multiplier, quantBytes) {
var toStr = { 1.0: '100', 0.75: '075', 0.50: '050' };
var graphJson = "model-stride" + stride + ".json";
if (quantBytes === 4) {
return MOBILENET_BASE_URL + ("float/" + toStr[multiplier] + "/") + graphJson;
}
else {
return MOBILENET_BASE_URL + ("quant" + quantBytes + "/" + toStr[multiplier] + "/") +
graphJson;
}
}
exports.mobileNetSavedModel = mobileNetSavedModel;
//# sourceMappingURL=saved_models.js.map