diff --git a/.gitignore b/.gitignore
index c3e07a9..82b0153 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,6 +5,7 @@ __pycache__/
 dist/
 build/
 *.egg
+*.whl
 
 # PyTorch
 *.pth
diff --git a/pyproject.toml b/pyproject.toml
index 3dd3bfc..f122459 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,6 +19,10 @@ dependencies = [
 ]
 
 # PyTorch CUDA版本配置
+
+[tool.setuptools.packages]
+find = {where = ["src"]}
+
 [tool.uv]
 # 指定PyTorch使用CUDA索引
 [[tool.uv.index]]
diff --git a/src/__init__.py b/src/infrared_data/__init__.py
similarity index 100%
rename from src/__init__.py
rename to src/infrared_data/__init__.py
diff --git a/src/detect.py b/src/infrared_data/detect.py
similarity index 100%
rename from src/detect.py
rename to src/infrared_data/detect.py
diff --git a/src/diagnose.py b/src/infrared_data/diagnose.py
similarity index 100%
rename from src/diagnose.py
rename to src/infrared_data/diagnose.py
diff --git a/src/extract_temp.py b/src/infrared_data/extract_temp.py
similarity index 100%
rename from src/extract_temp.py
rename to src/infrared_data/extract_temp.py
diff --git a/src/utils/__init__.py b/src/infrared_data/utils/__init__.py
similarity index 100%
rename from src/utils/__init__.py
rename to src/infrared_data/utils/__init__.py
diff --git a/src/utils/temp_utils.py b/src/infrared_data/utils/temp_utils.py
similarity index 100%
rename from src/utils/temp_utils.py
rename to src/infrared_data/utils/temp_utils.py