Блог

Главная - Вопросы по программированию - Как мне встроить диаграмму nnv в flask?

Как мне встроить диаграмму nnv в flask?

#python #flask #neural-network

#python #flask #нейронная сеть

Вопрос:

Я хочу встроить рендеринг nnv в flask, но мой код semms не отображает диаграмму нейронной сети.

Мой код:

 from nnv import NNV

layersList = [
    {"title":"input", "units": 3, "color": "darkBlue"},
    {"title":"hiddennlayer", "units": 3},
    {"title":"output", "units": 6,"color": "darkBlue"},
]




from flask import Flask

app = Flask(__name__)

@app.route('/')
def index():
  return f"{NNV(layersList).render()}"

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

Ответ №1:

Проблема в том, что matplotlib используется для построения графика нейронной сети; чтобы вернуть изображение в качестве HTTP-ответа, вам нужно преобразовать график в байты.

 import io
from flask import Flask, Response

@app.route('/')
def index():
    output = io.BytesIO()
    NNV(layersList).render(save_to_file=output)
    return Response(output.getvalue(), mimetype='image/png')