tensorflowjs初次摸底

最新推荐文章于 2024-02-26 09:30:00 发布

Maxwellhang

最新推荐文章于 2024-02-26 09:30:00 发布

阅读量176

点赞数

分类专栏： python web

本文链接：https://blog.csdn.net/u011529752/article/details/113921544

版权

TensorFlow.js 面部关键点检测模型预测预处理实时检测

关键词由CSDN通过智能技术生成

python 同时被 2 个专栏收录

17 篇文章 0 订阅

订阅专栏

web

1 篇文章 0 订阅

订阅专栏

安装tensorflowjs
npm install @tensorflow/tfjs
quant 1bit 要转成graph模型才行

<!DOCTYPE html>
<html>

<head>
    <script src="node_modules/@tensorflow/tfjs/dist/tf.min.js"></script>
</head>

<body>
    <img src="0_HELEN_111510095_1_4_0.png" width=112 height=112 id='img'>
    <script>
        var ldmks = null;
        var res = null;
        var inputTensor = null;
        async function run() {
            const imgElement = document.getElementById('img');
            const IMAGE_SIZE = 112;

            const model = await tf.loadLayersModel('checkpoint_robust_nopose_tfjs/model.json');
            // console.log(model);

            for (let i = 0; i < 10; ++i) {
                const startTime1 = performance.now();

                const img = tf.browser.fromPixels(imgElement).toFloat();

                const offset = tf.scalar(255.0);
                // Normalize the image from [0, 255] to [-1, 1].
                // const normalized = img.sub(offset).div(offset);
                const normalized = img.div(offset);
                // Reshape to a single-element batch so we can pass it to predict.
                const batched = normalized.reshape([1, IMAGE_SIZE, IMAGE_SIZE, 3]).transpose([0, 3, 1, 2]);
                inputTensor = batched;
                console.log(await batched.data());

                ldmks = await model.predict(batched);

                console.log(await ldmks.data());
                console.log(performance.now() - startTime1);
                res = await ldmks.data();
            }
        };
        run();
    </script>
</body>

</html>

Maxwellhang

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
tensorflowjs初次摸底

安装tensorflowjsnpm install @tensorflow/tfjsquant 1bit 要转成graph模型才行<!DOCTYPE html><html><head> <script src="node_modules/@tensorflow/tfjs/dist/tf.min.js"></script></head><body> <img src="0_HELEN.
复制链接

扫一扫