cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adds a public function for creating TaskRunner instances.

Browse Source 
PiperOrigin-RevId: 493109736
This commit is contained in:
MediaPipe Team 2022-12-05 14:14:13 -08:00 committed by Copybara-Service
parent 4f8eaee20f
commit 69b27b246a
1 changed files with 29 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
mediapipe/tasks/web/core/task_runner.ts
View File

@ -32,19 +32,13 @@ const GraphRunnerImageLibType =
/** An implementation of the GraphRunner that supports image operations */ /** An implementation of the GraphRunner that supports image operations */
export class GraphRunnerImageLib extends GraphRunnerImageLibType {} export class GraphRunnerImageLib extends GraphRunnerImageLibType {}
/** Base class for all MediaPipe Tasks. */ /**
export abstract class TaskRunner<O extends TaskRunnerOptions> {
protected abstract baseOptions: BaseOptionsProto;
protected graphRunner: GraphRunnerImageLib;
private processingErrors: Error[] = [];
/**
* Creates a new instance of a Mediapipe Task. Determines if SIMD is * Creates a new instance of a Mediapipe Task. Determines if SIMD is
* supported and loads the relevant WASM binary. * supported and loads the relevant WASM binary.
* @return A fully instantiated instance of `T`. * @return A fully instantiated instance of `T`.
*/ */
protected static async createInstance<T extends TaskRunner<O>, export async function
O extends TaskRunnerOptions>( createTaskRunner<T extends TaskRunner<O>, O extends TaskRunnerOptions>(
type: WasmMediaPipeConstructor<T>, initializeCanvas: boolean, type: WasmMediaPipeConstructor<T>, initializeCanvas: boolean,
fileset: WasmFileset, options: O): Promise<T> { fileset: WasmFileset, options: O): Promise<T> {
const fileLocator: FileLocator = { const fileLocator: FileLocator = {
@ -64,6 +58,24 @@ export abstract class TaskRunner<O extends TaskRunnerOptions> {
type, fileset.wasmLoaderPath, NO_ASSETS, canvas, fileLocator); type, fileset.wasmLoaderPath, NO_ASSETS, canvas, fileLocator);
await instance.setOptions(options); await instance.setOptions(options);
return instance; return instance;
}
/** Base class for all MediaPipe Tasks. */
export abstract class TaskRunner<O extends TaskRunnerOptions> {
protected abstract baseOptions: BaseOptionsProto;
protected graphRunner: GraphRunnerImageLib;
private processingErrors: Error[] = [];
/**
* Creates a new instance of a Mediapipe Task. Determines if SIMD is
* supported and loads the relevant WASM binary.
* @return A fully instantiated instance of `T`.
*/
protected static async createInstance<T extends TaskRunner<O>,
O extends TaskRunnerOptions>(
type: WasmMediaPipeConstructor<T>, initializeCanvas: boolean,
fileset: WasmFileset, options: O): Promise<T> {
return createTaskRunner(type, initializeCanvas, fileset, options);
} }
constructor( constructor(