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

Fix a crash that occurred when a model returns fewer vector elements than before

Browse Source 
PiperOrigin-RevId: 489041814
This commit is contained in:
Sebastian Schmidt 2022-11-16 14:36:14 -08:00 committed by Copybara-Service
parent 74474d859e
commit 3cdf0f6536
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mediapipe/web/graph_runner/wasm_mediapipe_lib.ts
View File

@ -406,7 +406,7 @@ export class WasmMediaPipeLib {
*/ */
setVectorListener<T>( setVectorListener<T>(
outputStreamName: string, callbackFcn: (data: T[]) => void) { outputStreamName: string, callbackFcn: (data: T[]) => void) {
const buffer: T[] = []; let buffer: T[] = [];
this.wasmModule.vectorListeners = this.wasmModule.vectorListeners || {}; this.wasmModule.vectorListeners = this.wasmModule.vectorListeners || {};
this.wasmModule.vectorListeners[outputStreamName] = this.wasmModule.vectorListeners[outputStreamName] =
(data: unknown, index: number, length: number) => { (data: unknown, index: number, length: number) => {
@ -419,6 +419,7 @@ export class WasmMediaPipeLib {
// the underlying data elements once we leave the scope of the // the underlying data elements once we leave the scope of the
// listener. // listener.
callbackFcn(buffer); callbackFcn(buffer);
buffer = [];
} }
}; };
} }