96 lines
3.4 KiB
Plaintext
96 lines
3.4 KiB
Plaintext
'''
|
||
Author: error: error: git config user.name & please set dead value or install git && error: git config user.email & please set dead value or install git & please set dead value or install git
|
||
Date: 2023-03-29 20:22:34
|
||
LastEditors: error: error: git config user.name & please set dead value or install git && error: git config user.email & please set dead value or install git & please set dead value or install git
|
||
LastEditTime: 2023-04-05 15:46:07
|
||
FilePath: /luyuetong-data/yolox-pytorch/flask_on.py
|
||
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://githaub.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
|
||
'''
|
||
from flask import Flask, make_response, request, jsonify,send_file
|
||
from flask_cors import CORS
|
||
from werkzeug.utils import secure_filename
|
||
from PIL import Image
|
||
from yolo import YOLO
|
||
from predict import PREDICT
|
||
import os
|
||
import quanju
|
||
import requests
|
||
import urllib.request
|
||
from io import BytesIO
|
||
import base64
|
||
import io
|
||
|
||
|
||
app = Flask(__name__)
|
||
CORS(app)
|
||
|
||
@app.route('/uploads', methods=['GET','POST'])
|
||
def uploads():
|
||
global filename
|
||
# 检查是否收到了POST请求
|
||
if request.method == 'POST':
|
||
|
||
|
||
# 检查是否收到了文件
|
||
if 'file' not in request.files:
|
||
return jsonify({'error': 'No file uploaded.'}), 400
|
||
|
||
file = request.files['file']
|
||
filename = file.read() #此时转换出来的是二进制内容
|
||
file_path = filename.decode("utf-8")
|
||
|
||
print(file_path)
|
||
print(f"file:{file}")
|
||
|
||
# 将blob转换为base64编码的字符串
|
||
image_data = base64.b64encode(file.read()).decode('utf-8')
|
||
|
||
# 将base64编码的字符串解码为图片数据
|
||
img = Image.open(io.BytesIO(base64.b64decode(image_data)))
|
||
Image.open(img)
|
||
# 将bytes类型传递给预测函数进行处理
|
||
result = PREDICT(img)
|
||
|
||
|
||
#
|
||
# print(file)
|
||
# filename = secure_filename(file.filename)
|
||
# file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
|
||
# result = PREDICT()
|
||
|
||
#检查是否选择了文件
|
||
if file.filename == '':
|
||
return jsonify({'error': 'No file selected.'}), 400
|
||
|
||
|
||
# # 检查文件类型是否合法
|
||
# # if file and allowed_file(file.filename):
|
||
# filename = secure_filename(file.filename)
|
||
#保存文件到本地
|
||
# file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
|
||
|
||
#自定义一个全局变量承接名字(自定义库.变量)
|
||
# quanju.mingzi = filename
|
||
return "ok"
|
||
|
||
# 将文件路径传递给模型进行预测
|
||
# result = PREDICT(os.path.join(app.config['UPLOAD_FOLDER'], filename))
|
||
|
||
# return send_file(f'./results/result_{quanju.mingzi}', mimetype='image/jpeg')
|
||
# return send_file('results/result_{}'.format(quanju.mingzi), mimetype='image/jpeg')
|
||
|
||
# return jsonify({'error': 'Invalid file type.'}), 400
|
||
|
||
|
||
|
||
def allowed_file(filename):
|
||
# 检查文件类型是否合法，这里仅接受jpeg、jpg、png类型的图片文件
|
||
return '.' in filename and filename.rsplit('.', 1)[1].lower() in {'jpeg', 'jpg', 'png'}
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# app.config['UPLOAD_FOLDER'] = './uploads'
|
||
app.config['UPLOAD_FOLDER'] = 'yolox-pytorch/uploads'
|
||
# app.debug = False
|
||
app.run(host='10.21.19.104', port=8000)
|