mirror of
https://github.com/luzhisheng/js_reverse.git
synced 2025-04-20 10:25:01 +08:00
30 lines
692 B
Python
30 lines
692 B
Python
from flask import Flask, request, jsonify
|
|
from detect import YOLOv5
|
|
from utils.general import cv2
|
|
import numpy as np
|
|
import base64
|
|
|
|
app = Flask(__name__)
|
|
yolov_5 = YOLOv5()
|
|
|
|
|
|
@app.route("/", methods=["POST"])
|
|
def index():
|
|
images = request.values.get("images")
|
|
input_image = base64.b64decode(images)
|
|
imBytes = np.frombuffer(input_image, np.uint8)
|
|
iImage = cv2.imdecode(imBytes, cv2.IMREAD_COLOR)
|
|
result = yolov_5.infer(iImage)
|
|
result = result[0].view(-1).int()
|
|
res_dict = {'coordinate': result.tolist()}
|
|
print(res_dict)
|
|
return jsonify(res_dict)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(
|
|
host='127.0.0.1',
|
|
port=8888,
|
|
debug=True
|
|
)
|