Add dataset QA and custom YOLO training flow

2026-06-30 14:04:11 +08:00
parent 43ed767b4f
commit 93af8bcd3a
14 changed files with 529 additions and 18 deletions
--- a/backend/app/schemas.py
+++ b/backend/app/schemas.py
@@ -47,3 +47,7 @@ class WeightSyncRequest(BaseModel):
 class DatasetCreate(BaseModel):
    name: str
    description: str = ""
+
+
+class DatasetYoloYamlRequest(BaseModel):
+    class_names: list[str] | None = None