Added train, val, test selctor

2023-03-22 21:21:44 +00:00
parent c15178da22
commit 51bd92d65a
8 changed files with 92 additions and 11 deletions
--- a/backend/src/models/signvideo.py
+++ b/backend/src/models/signvideo.py
@@ -9,6 +9,7 @@ class SignVideo(SQLModelExtended, table=True):
    id: int = Field(primary_key=True)

    approved: bool = False
+    dataset: str = "train" # train, test, val

    # foreign key to sign
    sign_id: int = Field(default=None, foreign_key="sign.id")
@@ -22,4 +23,5 @@ class SignVideo(SQLModelExtended, table=True):

 class SignVideoOut(BaseModel):
    id: int
-    approved: bool
+    approved: bool
+    dataset: str