ALVINN/ALVINN_f7
1
0
Fork 0
Code Issues 8 Pull Requests Actions Packages Projects Releases 7 Wiki Activity

Add performance tracking to local detection

Browse Source 
Signed-off-by: Justin Georgi <justin.georgi@gmail.com>
This commit is contained in:
Justin Georgi
2024-03-05 15:51:28 -07:00
parent 5086a52849
commit b00b5cf492
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/pages/local-detect.js
View File

@@ -10,13 +10,18 @@ export default {
}) })
}, },
async localDetect(imageData) { async localDetect(imageData) {
console.time('pre-process')
const [modelWidth, modelHeight] = model.inputs[0].shape.slice(1, 3); const [modelWidth, modelHeight] = model.inputs[0].shape.slice(1, 3);
const input = tf.tidy(() => { const input = tf.tidy(() => {
return tf.image.resizeBilinear(tf.browser.fromPixels(imageData), [modelWidth, modelHeight]).div(255.0).expandDims(0) return tf.image.resizeBilinear(tf.browser.fromPixels(imageData), [modelWidth, modelHeight]).div(255.0).expandDims(0)
}) })
console.timeEnd('pre-process')
console.time('run prediction')
const res = model.predict(input) const res = model.predict(input)
console.timeEnd('run prediction')
console.time('post-process')
const detectAttempts = res.shape[2] const detectAttempts = res.shape[2]
const outputSize = res.shape[1] const outputSize = res.shape[1]
const rawRes = tf.transpose(res,[0,2,1]).dataSync() const rawRes = tf.transpose(res,[0,2,1]).dataSync()
@@ -69,6 +74,7 @@ export default {
tf.dispose(res) tf.dispose(res)
tf.dispose(tBoxes) tf.dispose(tBoxes)
console.timeEnd('post-process')
return output return output
} }