admin/Seg_Data_Server_Net
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add dataset QA and custom YOLO training flow

Browse Source
This commit is contained in:
admin
2026-06-30 14:04:11 +08:00
parent 43ed767b4f
commit 93af8bcd3a
14 changed files with 529 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/app/catalog.py
View File

@@ -81,6 +81,7 @@ TASK_TYPES = [
"segmodel.metrics",
"segmodel.copy_best",
"yolo.train",
"yolo.train_custom",
"yolo.batch_train",
"yolo.predict",
"yolo.predict_v1",
@@ -181,6 +182,7 @@ TASK_DEFAULTS: dict[str, dict[str, Any]] = {
"segmodel.metrics": {},
"segmodel.copy_best": {},
"yolo.train": {"model": "YOLOv8n-seg"},
"yolo.train_custom": {"data": "var/uploads/datasets/example/dataset.yaml", "model": "YOLO11n-seg", "epochs": 10, "imgsz": 640, "batch": 1, "workers": 0, "device": "cpu", "exist_ok": True},
"yolo.batch_train": {},
"yolo.predict": {"model": "YOLOv8n-seg", "pt_name": "best.pt", "conf": 0.2, "run_choice": 1},
"yolo.predict_v1": {"model": "YOLOv8n-seg", "pt_name": "best.pt", "conf": 0.2, "run_choice": 1},