首页 > 解决方案 > 将图像从 Dropzone.js 加载到 Tensorflow.js

问题描述

目前,我们遇到了一个问题,我们需要允许用户将图像拖放到初始登录页面中,然后当他们按下Identify!按钮时,它会获取当前位于拖放图像位置的图像,然后将其扔到模型中然后被预测。

目前的问题是想办法把刚刚扔进Dropzone的图片放到我们的模型中进行预测。

let identify = document.querySelector('.identify')
if (identify) {
    identify.addEventListener('click', event => {
        event.preventDefault()
        let user_pic = document.querySelector('.dz-image').children[0].src
        ...
        const prediction = model.predict(user_pic)
        console.log('This is your prediction: ', prediction)
    })
}

正在发生的事情是控制台说: Uncaught TypeError: model.predict is not a function at HTMLDivElement.<anonymous>

这没有意义,因为我们的模板中有 CDN Tensorflow 标签,所以它应该可以正常工作。

标签: dropzone.jstensorflow.js

解决方案


您需要使用await,因为加载模型是异步操作

const model = await tf.loadLayersModel('model_json')

推荐阅读