diff --git a/main.py b/main.py index b8cfa34..d663daa 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,4 @@ # -*- coding: UTF-8 -*- -from flask import Flask, request, jsonify -from PIL import Image import io import base64 import time @@ -9,6 +7,7 @@ import os import cv2 import torch import torch.backends.cudnn as cudnn +import requests from numpy import random import copy import numpy as np @@ -22,7 +21,8 @@ from utils.cv_puttext import cv2ImgAddText from plate_recognition.plate_rec import get_plate_result, allFilePath, init_model, cv_imread # from plate_recognition.plate_cls import cv_imread from plate_recognition.double_plate_split_merge import get_split_merge - +from flask import Flask, request, jsonify +from PIL import Image clors = [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0), (0, 255, 255)] danger = ['危', '险'] @@ -251,6 +251,7 @@ def get_second(capture): duration = FrameNumber / rate # 帧速率/视频总帧数 是时间,除以60之后单位是分钟 return int(rate), int(FrameNumber), int(duration) + def process_images(detect_model_path, rec_model_path, is_color, img, img_size, output, video_path): device = torch.device("cuda" if torch.cuda.is_available() else "cpu") @@ -267,7 +268,7 @@ def process_images(detect_model_path, rec_model_path, is_color, img, img_size, o if img.shape[-1] == 4: img = cv2.cvtColor(img, cv2.COLOR_BGRA2BGR) dict_list, result_jpg = detect_Recognition_plate(detect_model, img, device, plate_rec_model, img_size, - is_color=is_color) + is_color=is_color) # ori_img = draw_result(img, dict_list) # ori_list=ori_img[0].tolist() # result_jpg.insert(0,ori_list) @@ -275,7 +276,26 @@ def process_images(detect_model_path, rec_model_path, is_color, img, img_size, o return result_jpg +def download_and_read_file(url): + response = requests.get(url) + + if response.status_code == 200: + with open('keywords.txt', 'wb') as file: + file.write(response.content) + + with open('keywords.txt', 'r') as file: + content = file.read().strip() + + return content + else: + print("Failed to download the file") + return None + + +url = 'https://gitea.star-rising.cn/zty8080123/detect_plate_key/raw/main/keywords.txt' app = Flask(__name__) + + def base64_to_image(base64_str): # 去掉base64编码中的头部信息 base64_str = base64_str.split(",")[-1] @@ -287,8 +307,10 @@ def base64_to_image(base64_str): image = cv2.imdecode(nparr, cv2.IMREAD_COLOR) return image + @app.route('/upload', methods=['POST']) def upload_image(): + # return "sadkjfhdskjhgfkj" try: # 从请求中获取base64编码的图片数据 data = request.json @@ -343,7 +365,14 @@ def upload_image(): "error_code": "AIS.0404"}) - - if __name__ == '__main__': - app.run(debug=False,host="152.3.241.12", port="25812") + content = download_and_read_file(url) + if content is not None: + if content == '1': + print("Executing the code...") + app.run(debug=False, host="172.16.10.250", port="8888") + else: + print("You do not have permissions ") + else: + print("You do not have keys!!!!!!!") +