64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
# python -m pip install flask
|
|
# export FLASK_APP=api.py
|
|
# flask run --without-threads
|
|
|
|
from flask import Flask, request, jsonify
|
|
from yolo import YOLO
|
|
import json
|
|
import yolo_video
|
|
import base64
|
|
|
|
app = Flask(__name__)
|
|
|
|
""" Automatic call while FLASK init """
|
|
|
|
# def deinit_yolo(yolo):
|
|
# yolo.close_session()
|
|
|
|
"""API_address/ping/<int:id>"""
|
|
@app.route("/ping/<int:id>", methods=["GET"])
|
|
def get_ping(id: int):
|
|
print(f'Request for PING {id=} received')
|
|
id = id + 1
|
|
return jsonify(id), 200
|
|
|
|
"""API_address/test"""
|
|
@app.get("/test")
|
|
def test():
|
|
o = {'str': 'okokok'}
|
|
return jsonify(o), 200
|
|
|
|
@app.route('/getPlatesNumber', methods=['POST'])
|
|
def getPlatesNumber():
|
|
yolo_model = YOLO()
|
|
res = json.loads(request.data)
|
|
|
|
# decoded data
|
|
decoded_data = base64.b64decode(res["imageFile"])
|
|
img_file = open('base64.png', 'wb')
|
|
img_file.write(decoded_data)
|
|
img_file.close()
|
|
image_path = r'base64.png'
|
|
base64_b, texts = yolo_video.detect_license_plate(model=yolo_model, img_path=image_path, i=0)
|
|
res = {'platesNumber': texts}
|
|
return jsonify(res), 200
|
|
|
|
"""API_address/detectLicense?img="""
|
|
@app.route("/detectLicense", methods=['POST'])
|
|
def detectLicensePlate():
|
|
yolo_model = YOLO()
|
|
res = json.loads(request.data)
|
|
|
|
# decoded data
|
|
decoded_data = base64.b64decode(res["imageFile"])
|
|
img_file = open('base64.png', 'wb')
|
|
img_file.write(decoded_data)
|
|
img_file.close()
|
|
image_path = r'base64.png'
|
|
base64_b, texts = yolo_video.detect_license_plate(model=yolo_model, img_path=image_path, i=0)
|
|
return jsonify(str(base64_b)), 200
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=8080)
|
|
|