利用flask进行图片数据传输

最新推荐文章于 2023-12-18 15:07:26 发布

cool_策

最新推荐文章于 2023-12-18 15:07:26 发布

阅读量2.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.csdn.net/qq_34510308/article/details/103008158

Python 专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一个使用Flask框架搭建的服务端应用，该应用能够接收客户端上传的图片，进行图像识别并返回识别结果。服务端通过加载预训练模型，对接收的图片进行解码和预测，最终将识别的类别和概率以JSON格式返回。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

服务端

from flask import request
from flask import Flask
import json
from predict import *
import cv2
import base64
from gevent.pywsgi import WSGIServer

app = Flask(__name__)


def load_model():
    predict = PREDICT(config)
    return predict


# 定义路由
@app.route("/photo", methods=['POST'])
def get_frame():
    # 接收图片
    # upload_file = json.loads(request.json)
    # print(upload_file['file'])
    # upload_file=request.form['file']
    filename=request.get_json()
    # print(filename)
    # print(type(filename)=='str')

    if (isinstance(filename,str)):
        filname1=json.loads(filename)
        tmp=filname1['file']
        img = base64.b64decode(str(tmp))
        image_data = np.fromstring(img, np.uint8)
        image_data = cv2.imdecode(image_data, cv2.IMREAD_COLOR)
        # cv2.imwrite('01.png', image_data)
    else:
        tmp=filename['file']
        img = base64.b64decode(str(tmp))
        image_data = np.fromstring(img, np.uint8)
        image_data = cv2.imdecode(image_data, cv2.IMREAD_COLOR)
        # cv2.imwrite('01.png', image_data)
    # fff=json.load(filename)
    # print(upload_file)
    # img=cv2.imread(tmp)
    # cv2.imwrite("222.jpg",img)
    className, prob = model.PredictImage(image_data)
    # print(className,prob)
    dic[className]=prob
    json_info = json.dumps(dic, ensure_ascii=False)
    return json_info


if __name__ == "__main__":
    dic = {}
    model=load_model()
    # print("初始化完成")
    app.run(host='0.0.0.0',port=5000)
    ## Serve the app with gevent
    # http_server = WSGIServer(('0.0.0.1',5000),app)
    # http_server.serve_forever()

客户端

import requests
import base64
import json
# API地址
url = "http://172.16.50.80:5000/photo"
# 图片地址
file_path = '/home/src/falsk/den.png'
# 图片名
# file_name = file_path.split('/')[-1]
# 二进制打开图片
# file = open(file_path, 'rb')
with open('D:/Image/baiban/test/den.png', 'rb') as f:
    img = base64.b64encode(f.read()).decode()
image = []
image.append(img)
# 拼接参数
files = {'file': image}
fff=json.dumps(files, ensure_ascii=False)
# 发送post请求到服务器端json.dumps(files, ensure_ascii=False)
# r = requests.post(url, json=json.dumps(files, ensure_ascii=False))
r = requests.post(url, json=fff)
print(r.status_code)
print(r.content)
print(type(json.dumps(files, ensure_ascii=False)))
# 获取服务器返回的图片，字节流返回
# result = r.content
# 字节转换成图片
# img = base64.b64decode(result)
# file = open('test.jpg', 'wb')
# file.write(img)
# file.close()