Include requirements for logging

chenzhaiyu · Nov 22, 2022 · 4face60 · 4face60
1 parent 4918a7a
commit 4face60
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -44,6 +44,12 @@ pip install -r points2surf/requirements.txt
 For training, make sure CUDA is available and enabled.
 Navigate to [`points2surf/README.md`](https://github.com/ErlerPhilipp/points2surf) for more details on its requirements.
 
+In addition, install dependencies for logging:
+
+```bash
+pip install -r requirements.txt
+```
+
 ## Getting started
 
 ### Reconstrction demo 

diff --git a/conf/config.yaml b/conf/config.yaml
@@ -43,9 +43,10 @@ eval_samples: 10000
 eval_csv: 'hausdorff_dist_pred_rec.csv'
 
 # hydra settings
-override hydra/job_logging: colorlog
-override hydra/hydra_logging: colorlog
 hydra:
   run:
     dir: '${outdir}'
   verbose: false
+defaults:
+  - override hydra/job_logging: colorlog
+  - override hydra/hydra_logging: colorlog
diff --git a/download.py b/download.py
@@ -43,7 +43,7 @@ def update_to(b=1, bsize=1, tsize=None):
     return update_to
 
 
-@hydra.main(config_path='./conf', config_name='config')
+@hydra.main(config_path='./conf', config_name='config', version_base='1.2')
 def download(cfg: DictConfig):
     """
     Download datasets and/or models from public urls.

diff --git a/evaluate.py b/evaluate.py
@@ -14,7 +14,7 @@
 from points2surf.source.base import evaluation
 
 
-@hydra.main(config_path='./conf', config_name='config')
+@hydra.main(config_path='./conf', config_name='config', version_base='1.2')
 def evaluate_hausdorff_dist(cfg: DictConfig):
     """
     Evaluate Hausdorff distance between reconstructed and GT models.

diff --git a/reconstruct.py b/reconstruct.py
@@ -29,7 +29,7 @@
 from utils import create_cell_complex, create_query_points, extract_surface, infer_sdf
 
 
-@hydra.main(config_path='./conf', config_name='config')
+@hydra.main(config_path='./conf', config_name='config', version_base='1.2')
 def reconstruct_full(cfg: DictConfig):
     """
     Full reconstruction pipeline starting from point clouds.
@@ -82,7 +82,7 @@ def reconstruct_full(cfg: DictConfig):
                         sdf_values, graph_cut=True, coefficient=cfg.coefficient)
 
 
-@hydra.main(config_path='./conf', config_name='config')
+@hydra.main(config_path='./conf', config_name='config', version_base='1.2')
 def reconstruct_surface(cfg: DictConfig):
     """
     Reconstruction of surface extraction from cell complexes and SDF predictions.

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+hydra-core==1.2.0
+hydra-colorlog==1.2.0
+omegaconf==2.2.3