[ADD][FEAT] Normalization method to config

wbenbihi · Aug 21, 2022 · 1d85ff5 · 1d85ff5
1 parent 5700530
commit 1d85ff5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/config/config.default.yaml b/config/config.default.yaml
@@ -55,6 +55,7 @@ dataset:
   output_size: 64
   stacks: 3
   heatmap_stddev: 5.
+  normalization: "FromZero"
   sets:
     test: true
     validation: true

diff --git a/hourglass_tensorflow/utils/config.py b/hourglass_tensorflow/utils/config.py
@@ -74,6 +74,7 @@ class HTFDatasetConfig(BaseModel):
     output_size: int = 64
     stacks: int = 4
     heatmap_stddev: float = 10.0
+    normalization: Optional[str] = None
     params: Optional[HTFDatasetParamsConfig] = Field(
         default_factory=HTFDatasetParamsConfig
     )