Ожидаемый input_1 будет иметь форму (224, 224, 3), но получил массив с формой (400, 401, 3)

Question

Я получаю следующую ошибку при вызове функции model.predict при запуске модели классификации текста в Keras.

Expected input_1 to have shape (224, 224, 3) but got array with shape (400, 401, 3)

Это код моей модели

from flask import render_template, jsonify, Flask, redirect, url_for, request
from app import app
import random
import os
from tensorflow.keras.applications.resnet50 import ResNet50
from tensorflow.keras.preprocessing import image
from tensorflow.keras.applications.resnet50 import preprocess_input, decode_predictions
import numpy as np

UPLOAD_FOLDER = '/Users/lorenzocastagno/Desktop/flaskSaaS-master/app/forms'

@app.route('/')
@app.route('/index')
def index():
     return render_template('index.html', title='Home')




@app.route('/upload.php', methods = ['GET', 'POST'])
def upload_file():
   if request.method == 'POST':
      f = request.files['file']
      path = os.path.join(app.config['UPLOAD_FOLDER'], f.filename)
      model= ResNet50(weights='imagenet')
      img = image.load_img(os.path.join(UPLOAD_FOLDER, f.filename))
      x = image.img_to_array(img)
      x = np.expand_dims(x, axis=0)
      x = preprocess_input(x)
      preds = model.predict(x)
      preds_decoded = decode_predictions(preds, top=3)[0] 
      print(decode_predictions(preds, top=3)[0])
      f.save(path)
      return render_template('uploaded.html', title='Success', predictions=preds_decoded, user_image=f.filename)





@app.route('/map')
def map():
    return render_template('map.html', title='Map')


@app.route('/map/refresh', methods=['POST'])
def map_refresh():
    points = [(random.uniform(48.8434100, 48.8634100),
               random.uniform(2.3388000, 2.3588000))
              for _ in range(random.randint(2, 9))]
    return jsonify({'points': points})


@app.route('/contact')
def contact():
    return render_template('contact.html', title='Contact')

Я не могу понять, где проблема и как ее исправить.

Я думаюпроблема заключается в измерении данных, я должен изменить его?

Answer 1

Вам необходимо изменить размер x до (224,224,3), потому что ваша модель принимает только вход этой формы.