完成推理模块的转移

This commit is contained in:
2025-04-17 15:57:16 +08:00
parent 74e8f0d415
commit b0379e64c9
130 changed files with 14269 additions and 3201 deletions

View File

@ -1,14 +1,13 @@
from . import schemas, models, crud
from apps.business.project import schemas as proj_schemas, models as proj_models, crud as proj_crud
from utils import os_utils as os
from application.settings import *
from . import schemas, models, crud
from utils.websocket_server import room_manager
from apps.business.project import models as proj_models, crud as proj_crud
import yaml
import asyncio
import subprocess
from typing import List
from redis.asyncio import Redis
from sqlalchemy.ext.asyncio import AsyncSession
@ -73,7 +72,7 @@ async def before_train(proj_info: proj_models.ProjectInfo, db: AsyncSession):
async def operate_img_label(
img_list: List[proj_models.ProjectImgLabel],
img_list: list[proj_models.ProjectImgLabel],
img_path: str,
label_path: str,
db: AsyncSession,