No public description

PiperOrigin-RevId: 564894013
This commit is contained in:
Sebastian Schmidt 2023-09-12 18:38:34 -07:00 committed by Copybara-Service
parent df211d211e
commit a5a3e9d36b

View File

@ -47,10 +47,15 @@ async function isSimdSupported(): Promise<boolean> {
} }
async function createFileset( async function createFileset(
taskName: MediapipeTaskCategory, basePath = ''): Promise<WasmFileset> { taskName: MediapipeTaskCategory, basePath?: string): Promise<WasmFileset> {
const suffix = const suffix =
await isSimdSupported() ? 'wasm_internal' : 'wasm_nosimd_internal'; await isSimdSupported() ? 'wasm_internal' : 'wasm_nosimd_internal';
// For backwards compatiblity, we treat an unset `basePath` as a relative
// path. FilesetResolver provides an empty path as a default, which is not
// rewritten to '.'.
basePath = basePath ?? '.';
return { return {
wasmLoaderPath: `${basePath}/${taskName}_${suffix}.js`, wasmLoaderPath: `${basePath}/${taskName}_${suffix}.js`,
wasmBinaryPath: `${basePath}/${taskName}_${suffix}.wasm`, wasmBinaryPath: `${basePath}/${taskName}_${suffix}.wasm`,
@ -93,7 +98,7 @@ export class FilesetResolver {
* @return A `WasmFileset` that can be used to initialize MediaPipe Audio * @return A `WasmFileset` that can be used to initialize MediaPipe Audio
* tasks. * tasks.
*/ */
static forAudioTasks(basePath?: string): Promise<WasmFileset> { static forAudioTasks(basePath = ''): Promise<WasmFileset> {
return createFileset('audio', basePath); return createFileset('audio', basePath);
} }
@ -107,7 +112,7 @@ export class FilesetResolver {
* @return A `WasmFileset` that can be used to initialize MediaPipe Text * @return A `WasmFileset` that can be used to initialize MediaPipe Text
* tasks. * tasks.
*/ */
static forTextTasks(basePath?: string): Promise<WasmFileset> { static forTextTasks(basePath = ''): Promise<WasmFileset> {
return createFileset('text', basePath); return createFileset('text', basePath);
} }
@ -121,7 +126,7 @@ export class FilesetResolver {
* @return A `WasmFileset` that can be used to initialize MediaPipe Vision * @return A `WasmFileset` that can be used to initialize MediaPipe Vision
* tasks. * tasks.
*/ */
static forVisionTasks(basePath?: string): Promise<WasmFileset> { static forVisionTasks(basePath = ''): Promise<WasmFileset> {
return createFileset('vision', basePath); return createFileset('vision', basePath);
} }
} }