btl143/project_infer
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
闫浩霖
2024-07-04 17:49:41 +08:00
parent 65aceec3a2
commit d81d0d7cff
3 changed files with 67 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
ocr_server/ocr_infer_server.py
View File

@@ -31,8 +31,8 @@ def get_access_token():
def ocr_api_request(image_base64):
# url = "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic" # 高精度
# url = "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate" # 高精度带位置
url = "https://aip.baidubce.com/rest/2.0/ocr/v1/general" # 标准精度带位置
url = "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate" # 高精度带位置
# url = "https://aip.baidubce.com/rest/2.0/ocr/v1/general" # 标准精度带位置
headers = {
'Content-Type': 'application/json'
}
@@ -58,16 +58,21 @@ def ocr_api_request(image_base64):
if __name__ == "__main__":
cfg = toml.load('../cfg_infer_server.toml')
cfg = toml.load('/home/evan/Workplace/project_infer/cfg_infer_server.toml')
# 配置日志输出
logger.add(cfg['debug']['logger_filename'], format=cfg['debug']['logger_format'], retention = 5, level="INFO")
context1 = zmq.Context()
camera_socket = context1.socket(zmq.REQ)
camera_socket.connect(f"tcp://localhost:{cfg['camera']['camera2_port']}")
logger.info("connect camera success")
# context1 = zmq.Context()
# camera_socket = context1.socket(zmq.REQ)
# camera_socket.connect(f"tcp://localhost:{cfg['camera']['camera2_port']}")
# logger.info("connect camera success")
cap = cv2.VideoCapture(20)
cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter.fourcc('M','J','P','G'))
cap.set(cv2.CAP_PROP_FPS, 20)
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 960)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 540)
# 初始化 server
context = zmq.Context()
@@ -75,27 +80,41 @@ if __name__ == "__main__":
socket = context.socket(zmq.REP)
socket.bind(f"tcp://*:{cfg['server']['ocr_infer_port']}")
import signal
import sys
def signal_handler(signum, frame):
logger.info("Received signal, exiting...")
socket.close()
context.term()
sys.exit(0)
# 注册信号处理函数
signal.signal(signal.SIGTERM, signal_handler)
signal.signal(signal.SIGINT, signal_handler)
while True:
message1 = socket.recv_string()
logger.info("recv client request")
for _ in range(5):
camera_socket.send_string("")
message = camera_socket.recv()
np_array = np.frombuffer(message, dtype=np.uint8)
image = cv2.imdecode(np_array, cv2.IMREAD_COLOR)
output_file_path = 'output_image.jpg'
success = cv2.imwrite(output_file_path, image)
encoded_image = base64.b64encode(message).decode('utf-8')
result = ocr_api_request(encoded_image)
print(result)
if result != None:
socket.send_pyobj({'code': 0, 'content': result.get('words_result')})
ret, frame = cap.read()
if ret:
frame = frame[:,0:480]
frame = cv2.rotate(frame, cv2.ROTATE_90_COUNTERCLOCKWISE)
output_file_path = 'rotate.jpg'
success = cv2.imwrite(output_file_path, frame)
_, frame = cv2.imencode('.jpg', frame)
encoded_image = base64.b64encode(frame).decode('utf-8')
result = ocr_api_request(encoded_image)
print(result)
if result != None:
socket.send_pyobj({'code': 0, 'content': result.get('words_result')})
else:
socket.send_pyobj({'code': -1, 'content': " ocr 没找到文字"})
else:
socket.send_pyobj({'code': -1, 'content': None})
socket.send_pyobj({'code': -1, 'content': "ocr 摄像头读取出错"})
if cv2.waitKey(1) == 27:
break
logger.info("ocr infer server exit")