hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

finish nbv_reconstruction_dataset

Browse Source
This commit is contained in:
hofee
2024-08-22 22:28:20 +08:00
parent 7cd1954a25
commit 954fed1122
5 changed files with 117 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
runners/strategy_generator.py
View File

@@ -7,14 +7,14 @@ import PytorchBoot.stereotype as stereotype
from utils.data_load import DataLoadUtil
from utils.reconstruction import ReconstructionUtil
@stereotype.runner("strategy_generator", comment="unfinished")
@stereotype.runner("strategy_generator")
class StrategyGenerator(Runner):
def __init__(self, config):
super().__init__(config)
self.load_experiment("generate")
def run(self):
dataset_name_list = ConfigManager.get("runner", "dataset_list")
dataset_name_list = ConfigManager.get("runner", "generate" "dataset_list")
voxel_threshold, overlap_threshold = ConfigManager.get("runner","generate","voxel_threshold"), ConfigManager.get("runner","generate","overlap_threshold")
for dataset_name in dataset_name_list:
root_dir = ConfigManager.get("datasets", dataset_name, "root_dir")