增加对训练和推理中图片情况的判断

2025-03-12 09:39:50 +08:00
parent f49a6caf10
commit 76f4a5ecd9
5 changed files with 51 additions and 1 deletions
--- a/app/service/project_detect_service.py
+++ b/app/service/project_detect_service.py
@ -120,7 +120,7 @@ def run_detect_yolo(detect_in: ProjectDetectLogIn, detect: ProjectDetect, train:
    return detect_log


-def run_commend(weights: str, source: str, project: str, name: str,
+async def run_commend(weights: str, source: str, project: str, name: str,
                log_id: int, detect_id: int, session: Session):
    """
    执行yolov5的推理