[ADD] Configure Input Shape

wbenbihi · Aug 20, 2022 · 00b1c7e · 00b1c7e
1 parent 3a3b7a9
commit 00b1c7e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/config/config.default.yaml b/config/config.default.yaml
@@ -54,6 +54,8 @@ dataset:
   sets:
     test: true
     validation: true
+  input:
+    size: 256
   split:
     column: set
     train_value: TRAIN

diff --git a/hourglass_tensorflow/utils/config.py b/hourglass_tensorflow/utils/config.py
@@ -68,12 +68,17 @@ class HTFDatasetBBoxConfig(BaseModel):
     )
 
 
+class HTFDatasetSizeConfig(BaseModel):
+    input: int = 256
+
+
 class HTFDatasetConfig(BaseModel):
     object: str
     params: Optional[HTFDatasetParamsConfig] = Field(
         default_factory=HTFDatasetParamsConfig
     )
     sets: Optional[HTFDatasetSetsConfig]
+    size: Optional[HTFDatasetSizeConfig] = Field(default_factory=HTFDatasetSizeConfig)
     split: Optional[HTFDatasetSplitConfig] = Field(
         default_factory=HTFDatasetSplitConfig
     )