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

Use uppercase options name for "delegate"

Browse Source 
PiperOrigin-RevId: 496764089
This commit is contained in:
Sebastian Schmidt 2022-12-20 15:15:24 -08:00 committed by Copybara-Service
parent a7b52d2c52
commit d2f738793c
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mediapipe/tasks/web/components/processors/base_options.test.ts
View File

@ -86,7 +86,7 @@ describe('convertBaseOptionsToProto()', () => {
it('can enable CPU delegate', async () => { it('can enable CPU delegate', async () => {
const baseOptionsProto = await convertBaseOptionsToProto({ const baseOptionsProto = await convertBaseOptionsToProto({
modelAssetBuffer: new Uint8Array(mockBytes), modelAssetBuffer: new Uint8Array(mockBytes),
delegate: 'cpu', delegate: 'CPU',
}); });
expect(baseOptionsProto.toObject()).toEqual(mockBytesResult); expect(baseOptionsProto.toObject()).toEqual(mockBytesResult);
}); });
@ -94,7 +94,7 @@ describe('convertBaseOptionsToProto()', () => {
it('can enable GPU delegate', async () => { it('can enable GPU delegate', async () => {
const baseOptionsProto = await convertBaseOptionsToProto({ const baseOptionsProto = await convertBaseOptionsToProto({
modelAssetBuffer: new Uint8Array(mockBytes), modelAssetBuffer: new Uint8Array(mockBytes),
delegate: 'gpu', delegate: 'GPU',
}); });
expect(baseOptionsProto.toObject()).toEqual({ expect(baseOptionsProto.toObject()).toEqual({
...mockBytesResult, ...mockBytesResult,
@ -117,7 +117,7 @@ describe('convertBaseOptionsToProto()', () => {
it('can reset delegate', async () => { it('can reset delegate', async () => {
let baseOptionsProto = await convertBaseOptionsToProto({ let baseOptionsProto = await convertBaseOptionsToProto({
modelAssetBuffer: new Uint8Array(mockBytes), modelAssetBuffer: new Uint8Array(mockBytes),
delegate: 'gpu', delegate: 'GPU',
}); });
// Clear backend // Clear backend
baseOptionsProto = baseOptionsProto =

2
mediapipe/tasks/web/components/processors/base_options.ts
View File

@ -71,7 +71,7 @@ async function configureExternalFile(
/** Configues the `acceleration` option. */ /** Configues the `acceleration` option. */
function configureAcceleration(options: BaseOptions, proto: BaseOptionsProto) { function configureAcceleration(options: BaseOptions, proto: BaseOptionsProto) {
const acceleration = proto.getAcceleration() ?? new Acceleration(); const acceleration = proto.getAcceleration() ?? new Acceleration();
if (options.delegate === 'gpu') { if (options.delegate === 'GPU') {
acceleration.setGpu(new InferenceCalculatorOptions.Delegate.Gpu()); acceleration.setGpu(new InferenceCalculatorOptions.Delegate.Gpu());
} else { } else {
acceleration.setTflite(new InferenceCalculatorOptions.Delegate.TfLite()); acceleration.setTflite(new InferenceCalculatorOptions.Delegate.TfLite());

2
mediapipe/tasks/web/core/task_runner_options.d.ts vendored
View File

@ -31,7 +31,7 @@ export declare interface BaseOptions {
modelAssetBuffer?: Uint8Array|undefined; modelAssetBuffer?: Uint8Array|undefined;
/** Overrides the default backend to use for the provided model. */ /** Overrides the default backend to use for the provided model. */
delegate?: 'cpu'|'gpu'|undefined; delegate?: 'CPU'|'GPU'|undefined;
} }
/** Options to configure MediaPipe Tasks in general. */ /** Options to configure MediaPipe Tasks in general. */