diff --git a/.gitignore b/.gitignore index 158fe5b..ef01995 100644 --- a/.gitignore +++ b/.gitignore @@ -167,4 +167,5 @@ cython_debug/ *.obj *.npz *.step -test_data/ \ No newline at end of file +test_data/ +logs/ \ No newline at end of file diff --git a/networks/__init__.py b/__init__.py similarity index 100% rename from networks/__init__.py rename to __init__.py diff --git a/utils/__init__.py b/brep2sdf/data/__init__.py similarity index 100% rename from utils/__init__.py rename to brep2sdf/data/__init__.py diff --git a/data/data.py b/brep2sdf/data/data.py similarity index 98% rename from data/data.py rename to brep2sdf/data/data.py index 54ef3fb..eec7244 100644 --- a/data/data.py +++ b/brep2sdf/data/data.py @@ -3,7 +3,7 @@ import torch from torch.utils.data import Dataset import numpy as np import pickle -from utils.logger import setup_logger +from brep2sdf.utils.logger import setup_logger # 设置日志记录器 logger = setup_logger('dataset') diff --git a/deep_sdf/__init__.py b/brep2sdf/deep_sdf/__init__.py similarity index 100% rename from deep_sdf/__init__.py rename to brep2sdf/deep_sdf/__init__.py diff --git a/deep_sdf/data.py b/brep2sdf/deep_sdf/data.py similarity index 100% rename from deep_sdf/data.py rename to brep2sdf/deep_sdf/data.py diff --git a/deep_sdf/mesh.py b/brep2sdf/deep_sdf/mesh.py similarity index 100% rename from deep_sdf/mesh.py rename to brep2sdf/deep_sdf/mesh.py diff --git a/deep_sdf/metrics/chamfer.py b/brep2sdf/deep_sdf/metrics/chamfer.py similarity index 100% rename from deep_sdf/metrics/chamfer.py rename to brep2sdf/deep_sdf/metrics/chamfer.py diff --git a/deep_sdf/utils.py b/brep2sdf/deep_sdf/utils.py similarity index 100% rename from deep_sdf/utils.py rename to brep2sdf/deep_sdf/utils.py diff --git a/deep_sdf/workspace.py b/brep2sdf/deep_sdf/workspace.py similarity index 100% rename from deep_sdf/workspace.py rename to brep2sdf/deep_sdf/workspace.py diff --git a/brep2sdf/networks/__init__.py b/brep2sdf/networks/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/networks/decoder.py b/brep2sdf/networks/decoder.py similarity index 100% rename from networks/decoder.py rename to brep2sdf/networks/decoder.py diff --git a/networks/encoder.py b/brep2sdf/networks/encoder.py similarity index 100% rename from networks/encoder.py rename to brep2sdf/networks/encoder.py diff --git a/networks/network.py b/brep2sdf/networks/network.py similarity index 100% rename from networks/network.py rename to brep2sdf/networks/network.py diff --git a/scripts/convert/brep_to_mesh.py b/brep2sdf/scripts/convert/brep_to_mesh.py similarity index 100% rename from scripts/convert/brep_to_mesh.py rename to brep2sdf/scripts/convert/brep_to_mesh.py diff --git a/scripts/convert/mesh_to_sdf.py b/brep2sdf/scripts/convert/mesh_to_sdf.py similarity index 100% rename from scripts/convert/mesh_to_sdf.py rename to brep2sdf/scripts/convert/mesh_to_sdf.py diff --git a/scripts/convert_utils.py b/brep2sdf/scripts/convert_utils.py similarity index 100% rename from scripts/convert_utils.py rename to brep2sdf/scripts/convert_utils.py diff --git a/scripts/process_brep.py b/brep2sdf/scripts/process_brep.py similarity index 100% rename from scripts/process_brep.py rename to brep2sdf/scripts/process_brep.py diff --git a/scripts/process_furniture.py b/brep2sdf/scripts/process_furniture.py similarity index 100% rename from scripts/process_furniture.py rename to brep2sdf/scripts/process_furniture.py diff --git a/scripts/read_pkl.py b/brep2sdf/scripts/read_pkl.py similarity index 100% rename from scripts/read_pkl.py rename to brep2sdf/scripts/read_pkl.py diff --git a/brep2sdf/utils/__init__.py b/brep2sdf/utils/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/utils/logger.py b/brep2sdf/utils/logger.py similarity index 100% rename from utils/logger.py rename to brep2sdf/utils/logger.py diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..674ae07 --- /dev/null +++ b/setup.py @@ -0,0 +1,7 @@ +from setuptools import setup, find_packages + +setup( + name="brep2sdf", + version="0.1.0", + packages=find_packages(), +) \ No newline at end of file