Showing
1 changed file
with
3 additions
and
2 deletions
... | @@ -61,7 +61,7 @@ async function predict() { | ... | @@ -61,7 +61,7 @@ async function predict() { |
61 | // Prediction 2: run input through teachable machine classification model | 61 | // Prediction 2: run input through teachable machine classification model |
62 | const prediction = await model.predict(posenetOutput); | 62 | const prediction = await model.predict(posenetOutput); |
63 | 63 | ||
64 | - if (prediction[0].probability.toFixed(2) > 0.9) { // 서있는 상태 | 64 | + if (prediction[0].probability.toFixed(2) > 0.90) { // 서있는 상태 |
65 | if (status == "squat"){ // 전에 스쿼트 상태였다면, 일어날 때 카운트를 하나 올려줘야 함. | 65 | if (status == "squat"){ // 전에 스쿼트 상태였다면, 일어날 때 카운트를 하나 올려줘야 함. |
66 | count++; | 66 | count++; |
67 | var audio = new Audio('./sound/' + count%10 + '.wav'); | 67 | var audio = new Audio('./sound/' + count%10 + '.wav'); |
... | @@ -70,9 +70,11 @@ async function predict() { | ... | @@ -70,9 +70,11 @@ async function predict() { |
70 | 70 | ||
71 | } | 71 | } |
72 | status = "stand" | 72 | status = "stand" |
73 | + console.log(status) | ||
73 | 74 | ||
74 | } else if (prediction[1].probability.toFixed(2) == 1.00) { // 스쿼트 자세 | 75 | } else if (prediction[1].probability.toFixed(2) == 1.00) { // 스쿼트 자세 |
75 | status = "squat" | 76 | status = "squat" |
77 | + console.log(status) | ||
76 | 78 | ||
77 | } else if (prediction[2].probability.toFixed(2) == 1.00) { // 굽은 자세(잘못된 케이스) | 79 | } else if (prediction[2].probability.toFixed(2) == 1.00) { // 굽은 자세(잘못된 케이스) |
78 | if (status == "squat" || status == "stand") { // 굽은 자세로 잘못 수행하면, 소리 나도록 | 80 | if (status == "squat" || status == "stand") { // 굽은 자세로 잘못 수행하면, 소리 나도록 |
... | @@ -81,7 +83,6 @@ async function predict() { | ... | @@ -81,7 +83,6 @@ async function predict() { |
81 | 83 | ||
82 | } | 84 | } |
83 | status = "bent" | 85 | status = "bent" |
84 | - console.log(status); | ||
85 | } | 86 | } |
86 | 87 | ||
87 | for (let i = 0; i < maxPredictions; i++) { | 88 | for (let i = 0; i < maxPredictions; i++) { | ... | ... |
-
Please register or login to post a comment