|
|
@@ -26,6 +26,12 @@ CLOUD_BASED_AI_URL = '34.192.147.108:8001'
|
|
|
MODEL_NAME = 'AI_5obj_pdcpv_detect_yolov5_pipeline'
|
|
|
# 建立长连接
|
|
|
ai_connect = AiEngine(url=CLOUD_BASED_AI_URL)
|
|
|
+# 检查连通性、推理服务器状态
|
|
|
+if ai_connect.health:
|
|
|
+ LOGGING.info('健康状况通过')
|
|
|
+# 设定模型
|
|
|
+if ai_connect.set_model(MODEL_NAME):
|
|
|
+ LOGGING.info('设置模型通过')
|
|
|
|
|
|
|
|
|
class AiView(View):
|
|
|
@@ -225,20 +231,12 @@ class AiView(View):
|
|
|
"""
|
|
|
try:
|
|
|
|
|
|
- # 检查连通性、推理服务器状态
|
|
|
- if ai_connect.health:
|
|
|
- LOGGING.info('健康状况通过')
|
|
|
- # 设定模型
|
|
|
- if ai_connect.set_model(MODEL_NAME):
|
|
|
- LOGGING.info('设置模型通过')
|
|
|
# 推理张数(一次最多推理128张！)
|
|
|
# 图片名称（这里可以改成内存）注意改完之后要检查input_tmp的【类型(type)、尺寸(shape)】是否和之前的一致
|
|
|
# 输入尺寸固定640wx360h，如需变动可以联系我们，我们这边做resize会快
|
|
|
input_name_arr = np.array(list(map(np.array, map(Image.open, input_name_arr))))
|
|
|
# 推理
|
|
|
results = ai_connect.yolo_infer(input_name_arr, nms_threshold, confidence, client_timeout)
|
|
|
- # 推理完请关闭长连接
|
|
|
- ai_connect.close()
|
|
|
# 报错返回
|
|
|
if results == 'e_timeout':
|
|
|
raise Exception('推理超时')
|
zhangdongming