Skip to content

GPU Acceleration to increase performance #377

Closed
@milind-yadav

Description

@milind-yadav

Do we have any option to use GPU to increase performance of model loading and detection?
As currently in Object Detection it's taking around 10 seconds. If we want to do this on GPU, can we do that?

Running below lines through web worker, increases overall UI experience but not increases any performance.

const model = await pipeline("object-detection", "Xenova/detr-resnet-50");
const result = await model(img, { threshold: 0.9 });

Can we use GPU for that?

Metadata

Metadata

Assignees

No one assigned

    Labels

    questionFurther information is requested

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions