jhynsoo/worlabel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'ai/develop' of https://lab.ssafy.com/s11-s-project/S11P21S002 into ai/refactor/resource-cleanup

Browse Source
This commit is contained in:
김진현 2024-10-04 15:43:05 +09:00
commit dae4ca4bf1
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ai/app/api/yolo/detection.py
View File

@ -51,7 +51,7 @@ async def detection_predict(request: PredictRequest):
send_slack_message(f"추론 결과 변환 완료 (projectId: {project_id}). 걸린 시간: {time.time() - start_time:.2f} 초", send_slack_message(f"추론 결과 변환 완료 (projectId: {project_id}). 걸린 시간: {time.time() - start_time:.2f} 초",
status="success") status="success")
send_slack_message(f"Detection predict 성공 (projectId: {project_id}) {response}", status="success") send_slack_message(f"Detection predict 성공 (projectId: {project_id}) {len(response)}", status="success")
return response return response

2
ai/app/api/yolo/segmentation.py
View File

@ -18,7 +18,7 @@ router = APIRouter()
@router.post("/predict") @router.post("/predict")
async def segmentation_predict(request: PredictRequest): async def segmentation_predict(request: PredictRequest):
project_id = request.project_id project_id = request.project_id
send_slack_message(f"Segmentation predict 요청 (projectId: {project_id}, 이미지 개수: {len(request.data)})", send_slack_message(f"Segmentation predict 요청 (projectId: {project_id}, 이미지 개수: {len(request.image_list)})",
status="success") status="success")
# 모델 로드 # 모델 로드