新增底模

app.py

@@ -183,6 +183,7 @@ class TrainParams(BaseModel):
     """
     用于接收/yolov12/train接口的训练参数，所有参数均需前端传入。
     """
+    model: str          # 训练底模
     data: str           # 数据集配置文件路径
     epochs: int         # 训练轮数
     batch: int          # 批次大小
@@ -207,6 +208,7 @@ def yolov12_train(params: TrainParams):
     logging.info(f"请求参数: {params}")
     try:
         model = YOLO("yolov12.yaml")  # 如有yolov12n.yaml可替换
+        model.load(params.model)
         logging.info("开始模型训练...")
         results = model.train(
             data=params.data,