s478839/wko-projekt
2
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

last changes to API

Browse Source
This commit is contained in:
Szymon Parafiński 2023-02-01 01:09:19 +01:00
parent c1a4151034
commit e163d25776
4 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
API/api.py
View File

@ -1,5 +1,5 @@
# python -m pip install flask
# export FLASK_APP=api.py
# export FLASK_APP=API/api.py
# flask run --without-threads
from flask import Flask, request, jsonify
@ -9,11 +9,8 @@ import yolo_video
import base64
app = Flask(__name__)
yolo_model = YOLO()
""" Automatic call while FLASK init """
# def deinit_yolo(yolo):
# yolo.close_session()
"""API_address/ping/<int:id>"""
@app.route("/ping/<int:id>", methods=["GET"])
@ -30,15 +27,14 @@ def test():
@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 = open('API/base64.png', 'wb')
img_file.write(decoded_data)
img_file.close()
image_path = r'base64.png'
image_path = r'API/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
@ -46,18 +42,20 @@ def getPlatesNumber():
"""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 = open('API/base64.png', 'wb')
img_file.write(decoded_data)
img_file.close()
image_path = r'base64.png'
image_path = r'API/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)
# if __name__ == '__main__':
# app.run(threaded=False, host='0.0.0.0', port=8080)
### to run api just paste following to terminal from root dir of project
# export FLASK_APP=API/api.py
# flask run --without-threads

1
API/model_data/_classes.txt Normal file
View File

@ -0,0 +1 @@
licence

2
API/yolo.py
View File

@ -26,7 +26,7 @@ class YOLO(object):
_defaults = {
"model_path": 'API/model_data/trained_weights_final.h5',
"anchors_path": 'API/model_data/yolo_anchors.txt',
"classes_path": 'API/train/_classes.txt',
"classes_path": 'API/model_data/_classes.txt',
"score": 0.3,
"iou": 0.45,
"model_image_size": (256,256),#(128, 128),

2
API/yolo3/model.py
View File

@ -11,7 +11,7 @@ from keras.layers import BatchNormalization
from keras.models import Model
from keras.regularizers import l2
from API.yolo3.utils import compose
from yolo3.utils import compose
@wraps(Conv2D)