The example is ported from tfjs-examples/gpu-pipeline https://github.com/tensorflow/tfjs-examples/tree/master/gpu-pipeline, and could run on electron app based on tfjs webgpu backend. This demo shows a high performance webgpu pipeline with TF.js. In this example, the pipeline takes a camera stream as input, then feeds to a segmentation model from tfjs-models. The output of the model is then fed to a mask processing step to mask the background with purple. And then the result is painted on the canvas. For details on how to keep the tensor data on GPU, see our optimization doc.
-
Notifications
You must be signed in to change notification settings - Fork 0
xhcao/webgpu-pipeline-electron
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published