From e5d9855941d8f4772dd6b8c4e9a5b1f725e79b3f Mon Sep 17 00:00:00 2001 From: mckay Date: Wed, 13 Nov 2024 20:41:54 +0800 Subject: [PATCH] fix: modules import --- .gitignore | 3 ++- networks/__init__.py => __init__.py | 0 {utils => brep2sdf/data}/__init__.py | 0 {data => brep2sdf/data}/data.py | 2 +- {deep_sdf => brep2sdf/deep_sdf}/__init__.py | 0 {deep_sdf => brep2sdf/deep_sdf}/data.py | 0 {deep_sdf => brep2sdf/deep_sdf}/mesh.py | 0 {deep_sdf => brep2sdf/deep_sdf}/metrics/chamfer.py | 0 {deep_sdf => brep2sdf/deep_sdf}/utils.py | 0 {deep_sdf => brep2sdf/deep_sdf}/workspace.py | 0 brep2sdf/networks/__init__.py | 0 {networks => brep2sdf/networks}/decoder.py | 0 {networks => brep2sdf/networks}/encoder.py | 0 {networks => brep2sdf/networks}/network.py | 0 {scripts => brep2sdf/scripts}/convert/brep_to_mesh.py | 0 {scripts => brep2sdf/scripts}/convert/mesh_to_sdf.py | 0 {scripts => brep2sdf/scripts}/convert_utils.py | 0 {scripts => brep2sdf/scripts}/process_brep.py | 0 {scripts => brep2sdf/scripts}/process_furniture.py | 0 {scripts => brep2sdf/scripts}/read_pkl.py | 0 brep2sdf/utils/__init__.py | 0 {utils => brep2sdf/utils}/logger.py | 0 setup.py | 7 +++++++ 23 files changed, 10 insertions(+), 2 deletions(-) rename networks/__init__.py => __init__.py (100%) rename {utils => brep2sdf/data}/__init__.py (100%) rename {data => brep2sdf/data}/data.py (98%) rename {deep_sdf => brep2sdf/deep_sdf}/__init__.py (100%) rename {deep_sdf => brep2sdf/deep_sdf}/data.py (100%) rename {deep_sdf => brep2sdf/deep_sdf}/mesh.py (100%) rename {deep_sdf => brep2sdf/deep_sdf}/metrics/chamfer.py (100%) rename {deep_sdf => brep2sdf/deep_sdf}/utils.py (100%) rename {deep_sdf => brep2sdf/deep_sdf}/workspace.py (100%) create mode 100644 brep2sdf/networks/__init__.py rename {networks => brep2sdf/networks}/decoder.py (100%) rename {networks => brep2sdf/networks}/encoder.py (100%) rename {networks => brep2sdf/networks}/network.py (100%) rename {scripts => brep2sdf/scripts}/convert/brep_to_mesh.py (100%) rename {scripts => brep2sdf/scripts}/convert/mesh_to_sdf.py (100%) rename {scripts => brep2sdf/scripts}/convert_utils.py (100%) rename {scripts => brep2sdf/scripts}/process_brep.py (100%) rename {scripts => brep2sdf/scripts}/process_furniture.py (100%) rename {scripts => brep2sdf/scripts}/read_pkl.py (100%) create mode 100644 brep2sdf/utils/__init__.py rename {utils => brep2sdf/utils}/logger.py (100%) create mode 100644 setup.py 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