大小(27900)必须与形状1,224,224,3的乘积相匹配

我正在将包含裁剪面部的张量从BlazeFace传递到我的自定义模型中进行分类任务。问题在于这些维度与我的自定义模型的输入不匹配,我的模型是使用迁移学习和MobileNetV2训练的。我意识到当裁剪的面部尺寸小于我的自定义模型的形状时,我尝试调整它们的尺寸。我该怎么做才能让裁剪的面部适合模型呢?

模型输入形状 : [1,224,224,3]裁剪的面部 : [1,117,68,3]
        const crop = tf.image.cropAndResize(          tensorReshaped,          boxes,          [0],          [height, width]        );        // 将裁剪的面部调整大小到 [1,224,224,3]        console.log(crop.shape);        const prediction = model.predict(crop.reshape([1, 224, 224, 3]));        if (!prediction || prediction.length === 0) {          console.log("No prediction available");          return;        }        // 进行预测。        const preds = prediction.dataSync();        console.log(preds);

回答:

我需要调整裁剪图像的大小而不是改变它们的形状

const alignCorners = false;        // 将裁剪的图像调整大小到 [1,224,224,3]        const imageResize = tf.image.resizeBilinear(          crop,          [224, 224],          alignCorners        );

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注