face-api/build/common/getModelUris.js

28 lines
1.1 KiB
JavaScript
Raw Normal View History

2020-08-20 02:10:42 +02:00
export function getModelUris(uri, defaultModelName) {
2020-08-18 14:04:33 +02:00
const defaultManifestFilename = `${defaultModelName}-weights_manifest.json`;
if (!uri) {
return {
modelBaseUri: '',
manifestUri: defaultManifestFilename
};
}
if (uri === '/') {
return {
modelBaseUri: '/',
manifestUri: `/${defaultManifestFilename}`
};
}
const protocol = uri.startsWith('http://') ? 'http://' : uri.startsWith('https://') ? 'https://' : '';
uri = uri.replace(protocol, '');
const parts = uri.split('/').filter(s => s);
const manifestFile = uri.endsWith('.json')
? parts[parts.length - 1]
: defaultManifestFilename;
let modelBaseUri = protocol + (uri.endsWith('.json') ? parts.slice(0, parts.length - 1) : parts).join('/');
modelBaseUri = uri.startsWith('/') ? `/${modelBaseUri}` : modelBaseUri;
return {
modelBaseUri,
manifestUri: modelBaseUri === '/' ? `/${manifestFile}` : `${modelBaseUri}/${manifestFile}`
};
}
//# sourceMappingURL=getModelUris.js.map