From 83483dde35b8f349be4a542f85c67245787548ef Mon Sep 17 00:00:00 2001
From: swittl <simon.wittl@th-deg.de>
Date: Mon, 4 Nov 2024 16:37:05 +0100
Subject: [PATCH] chg
---
scripts/06_generate_dataset.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/scripts/06_generate_dataset.py b/scripts/06_generate_dataset.py
index d3fb72f..79f8dc4 100644
--- a/scripts/06_generate_dataset.py
+++ b/scripts/06_generate_dataset.py
@@ -11,7 +11,9 @@ and needs 100gb of space.
def main():
# Dataset save folder
- save_folder = Path(r'C:\data\XrayTransformLow')
+ save_folder = Path(r'\\192.168.43.8\data\Wittl\06_datasets\XrayTransform')
+ number_of_maps = 4
+ map_bounding_box = 25
# Dataset of multiple random stl files from: https://ten-thousand-models.appspot.com/
load_folder = Path(r'C:\Users\swittl\Downloads\Thingi10K\Thingi10K\raw_meshes')
@@ -22,7 +24,7 @@ def main():
print(i)
key = random.PRNGKey(i)
generate_atlas_from_folder(
- load_folder, key, save_folder, map_bounding_box=(-10, 10), number_of_maps=2)
+ load_folder, key, save_folder, map_bounding_box=(-map_bounding_box, map_bounding_box), number_of_maps=number_of_maps)
if __name__ == '__main__':
--
GitLab