From 00b1c7e8bb06f1a9f4a8ae01d4575d450521b2ac Mon Sep 17 00:00:00 2001
From: wbenbihi <waligoo@gmail.com>
Date: Sat, 20 Aug 2022 14:57:26 +0800
Subject: [PATCH] [ADD] Configure Input Shape

---
 config/config.default.yaml           | 2 ++
 hourglass_tensorflow/utils/config.py | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/config/config.default.yaml b/config/config.default.yaml
index 9dc8aa6..4794131 100644
--- 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
index 32ce29e..711be85 100644
--- 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
     )