Skip to content

Commit 88dc976

Browse files
authored
Merge pull request #38 from duoluodexiaoxiaoyuan/patch-1
2 parents 509ab4c + de90b23 commit 88dc976

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/page/tensorflow/pose/pose-detection.vue

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import { PoseDetector } from '@tensorflow-models/pose-detection'
33
import * as poseDetection from '@tensorflow-models/pose-detection'
44
import '@tensorflow/tfjs-backend-webgl'
5+
import * as tf from '@tensorflow/tfjs-core';
56
67
let loading = $ref(false)
78
let posenetInput: HTMLVideoElement | HTMLImageElement | HTMLCanvasElement
@@ -13,6 +14,8 @@ let model: poseDetection.SupportedModels.PoseNet
1314
// const model = poseDetection.SupportedModels.MoveNet
1415
// 初始化
1516
const init = async () => {
17+
await tf.setBackend('webgl');
18+
await tf.ready();
1619
loading = true
1720
// 获取 canvas 元素
1821
posenetOutput = document.getElementById('output') as HTMLCanvasElement

0 commit comments

Comments
 (0)