debug pose_diff

2024-09-27 08:06:49 +00:00
parent 030bf55192
commit 92250aeb62
3 changed files with 11 additions and 11 deletions
--- a/core/nbv_dataset.py
+++ b/core/nbv_dataset.py
@@ -34,7 +34,7 @@ class NBVReconstructionDataset(BaseDataset):
            self.model_dir = config["model_dir"]
            self.filter_degree = config["filter_degree"]
        if self.type == namespace.Mode.TRAIN:
-            scale_ratio = 100
+            scale_ratio = 1
            self.datalist = self.datalist*scale_ratio
        if self.cache:
            expr_root = ConfigManager.get("runner", "experiment", "root_dir")
@@ -83,7 +83,6 @@ class NBVReconstructionDataset(BaseDataset):
                            "label_idx": seq_idx,
                            "scene_max_coverage_rate": scene_max_coverage_rate
                        })
-                    break # TODO: for small version debug
        return datalist

    def preprocess_cache(self):