Я использую Tensorflow, чтобы распознать изображение.для этого я конвертировал это изображение в тензор и попытался обучить его и хотел сохранить эту модель и сделать прогноз на основе одного входного изображения.
Ниже приведен мой код, который я использовал, но застрял в этомошибка, которая может появиться во время обучения.
<apex:page id="PageId" showheader="false">
<head>
<title>Image Classifier with TensorFlowJS</title>
</head>
<body>
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.15.1">
</script>
</body>
<img id="imgshow" src="{!$Resource.cat}" crossorigin="anonymous" width="224" height="224" />
<button onClick="learnlinear()" lable = "Predict">Predict</button>
<script>
async function learnlinear()
{
//img data set
const imageHTML = document.getElementById('imgshow');
console.log('imageHTML::'+imageHTML.src);
//convert to tensor
const tensorImg = tf.fromPixels(imageHTML);
tensorImg.data().then(async function (stuffTensImg){
console.log('stuffTensImg::'+stuffTensImg.toString());
});
const resize_image = tf.reshape(tensorImg, [1, 224, 224, 3],'resize');
console.log('resize_image'+resize_image);
const model = await tf.loadModel('https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_0.25_224/model.json');
console.log(':::::::'+JSON.stringify(model));
// Use ADAM optimizer with learning rate of 0.0005 and MSE loss
model.compile({
optimizer: tf.train.adam(0.0005),
loss: 'meanSquaredError',
});
console.log('model.compile',model.compile);
await model.fit(resize_image, {epochs: 500});
console.log('model.fit',model.fit);
const saveResult = await model.save();
console.log('saveResult',saveResult);
model.predict(resize_image.expandDims(0)).print()
console.log('model.predict',model.predict);
}
</script>
Ошибка
Пожалуйста, помогите мне решить эту проблему или как я могу получитьэто.