|
|
@@ -26,8 +26,10 @@ 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)
|
|
|
+HEALTH = False
|
|
|
# 检查连通性、推理服务器状态
|
|
|
if ai_connect.health:
|
|
|
+ HEALTH = True
|
|
|
LOGGING.info('健康状况通过')
|
|
|
# 设定模型
|
|
|
if ai_connect.set_model(MODEL_NAME):
|
|
|
@@ -230,7 +232,9 @@ class AiView(View):
|
|
|
:return: results 推理结果
|
|
|
"""
|
|
|
try:
|
|
|
-
|
|
|
+ if not HEALTH:
|
|
|
+ LOGGING.info('AI health:{}'.format(HEALTH))
|
|
|
+ return {}
|
|
|
# 推理张数(一次最多推理128张！)
|
|
|
# 图片名称（这里可以改成内存）注意改完之后要检查input_tmp的【类型(type)、尺寸(shape)】是否和之前的一致
|
|
|
# 输入尺寸固定640wx360h，如需变动可以联系我们，我们这边做resize会快
|
zhangdongming