проблема с изображением fastai (ошибка типа: индексы байтов должны быть целыми или кусочками, а не str) - PullRequest
Новая
       115

проблема с изображением fastai (ошибка типа: индексы байтов должны быть целыми или кусочками, а не str)

0 голосов
/ 18 марта 2020

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

Вот код в вопросах: 

from fastai import *
from fastai.vision import *

path = Path('')
model = load_learner(path)

def get_veggie(image_bytes):
    img_bytes = image_bytes['file'].read()
    img = open_image(BytesIO(img_bytes))
    prediction = model.predict(img)[0]
    print(prediction)

И это app.py part: 

from flask import Flask, render_template, request
from commons import get_tensor
from inference import get_veggie

app = Flask(__name__)

@app.route('/', methods=['GET', 'POST'])
def hello_world():
    if request.method == 'GET':
        return render_template('index.html', value='O')
    if request.method == 'POST':
        if 'file' not in request.files:
            print('image not uploaded')
            return
        file = request.files['file']
        image = file.read()
        get_veggie(image_bytes=image)
        return render_template('result.html', veggie='cucumber')

if __name__ == '__main__':
    app.run(debug=True)

Все, что я смог найти по этому поводу, это темы о том, как десериализовать файлы json, что, похоже, не является проблемой здесь.

...