我已经在 CIFAR10 数据集(占位符,稍后将替换为不同的模型)上训练了 CNN,并将模型集成到 flask API 中。 API 托管在 Heroku 上,我现在想使用 Flutter/Dart 在我的手机上拍照,将它们发送到 Flask API,在它们上运行我训练过的模型并返回预测。
使用 python,我可以轻松地向我的 API 发出 post 请求并返回预测。这是我的简单python代码:
import requests
import json
img = open('some-picture.jpg', 'rb')
files = {'image': img}
response = requests.post("url_to_api", files=files)
print(response.text)
我很久没有使用 Flutter/Dart 了,我认为发出 htpp 请求的过程比在 python 中要复杂一些。有人可以给我一些指示或代码,让我可以用相机拍照,将其上传到我的 API,并将响应存储在变量中吗?这是 flask API 的(简化的)python 代码:
from flask import Flask, request
import os
import numpy as np
from PIL import Image
from tensorflow import keras
app = Flask(__name__)
app.config["DEBUG"] = True
model = keras.models.load_model('cifar10_cnn.h5')
labels = ["Airplane", "Automobile", "Bird", "Cat", "Deer", "Dog", "Frog", "Horse", "Ship", "Truck"]
@app.route('/', methods=["POST"])
def predict():
# stuff not relevant to question, left out for conciseness #
file = request.files['image']
image = Image.open(file).resize((32, 32))
image = np.array(image)
image = image / 255
image = image.reshape(-1, 32, 32, 3)
predictions = model.predict([image])
index = np.argmax(predictions)
results = {'Prediction:': labels[index]}
return results
if __name__ == '__main__':
app.run()
到目前为止,我知道 Multipart 文件似乎是要走的路,而且 Dio 包可能值得研究。如果可以提供进一步的提示或代码,我将不胜感激。
最佳答案
您可能已经知道如何从图库/相机中选择图像(例如,使用 image_picker
库)。您填写类字段,如 File image;
与该选择器的结果。这可能很简单:
import 'dart:io';
import 'package:image_picker/image_picker.dart';
class _MyHomePageState extends State<MyHomePage> {
File image;
final picker = ImagePicker();
pickImageFromGallery(ImageSource source) async {
final image = await picker.getImage(source: source);
setState(() {
this.image = File(image.path);
});
}
}
(更改 ImageSource source
以匹配您的需求:相机或画廊)然后您可以将该文件上传到您的 api。
使用
http
图书馆:import 'package:http/http.dart' as http;
import 'package:http_parser/http_parser.dart';
class _MyHomePageState extends State<MyHomePage> {
doUpload(){
var request = http.MultipartRequest(
'POST',
Uri.parse("url_to_api"),
);
Map<String, String> headers = {"Content-type": "multipart/form-data"};
request.files.add(
http.MultipartFile(
'image',
image.readAsBytes().asStream(),
image.lengthSync(),
filename: "filename",
contentType: MediaType('image', 'jpeg'),
),
);
request.headers.addAll(headers);
print("request: " + request.toString());
request.send().then((value) => print(value.statusCode));
}
}
对于这些库中的每一个,您必须将它们作为依赖项添加到 pubspec.yaml
中。在你的 flutter 项目中:cupertino_icons: ^0.1.3
http: ^0.12.2
image_picker: ^0.6.7
关于api - 使用 Flutter 向 Flask API 发送 http Post-Request(包含图像),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/64196318/