From ed2778972bcc13582c136d4a5b4ba7980326c390 Mon Sep 17 00:00:00 2001
From: yukun-hh <zhangyukunhh@gmail.com>
Date: Sun, 17 May 2026 19:51:50 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=B2=BE=E7=AE=80=E7=89=88?=
 =?UTF-8?q?=20requirements.txt=EF=BC=8C=E6=9B=B4=E6=96=B0=20README=20?=
 =?UTF-8?q?=E7=8E=AF=E5=A2=83=E8=A6=81=E6=B1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore       |  1 +
 README.md        | 15 +++------------
 requirements.txt | 10 ++++++++++
 3 files changed, 14 insertions(+), 12 deletions(-)
 create mode 100644 requirements.txt

diff --git a/.gitignore b/.gitignore
index eec9fbf..c650b04 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,6 +5,7 @@
 !/Merge_classes.py
 !/Model.py
 !/README.md
+!/requirements.txt
 !/THIRD_PARTY_LICENSES.md
 !/Train.py
 !/Baseline.py
diff --git a/README.md b/README.md
index ac9d41a..27f6a86 100644
--- a/README.md
+++ b/README.md
@@ -75,18 +75,9 @@
 
 ## 环境要求
 
-本项目无 `requirements.txt`，需手动安装以下依赖：
-
-- Python 3.8+
-- PyTorch（推荐 1.10+）
-- torchvision
-- tqdm
-- matplotlib
-- pandas
-- Pillow
-- torchsummary
-- scikit-learn（仅 `Evaluate.py` 需要）
-- scikit-image（仅 `baseline/HOG_Baseline.py` 需要）
+```bash
+pip install -r requirements.txt
+```
 
 ## 快速开始
 
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..f612349
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,10 @@
+torch>=1.10
+torchvision>=0.11
+tqdm
+matplotlib
+pandas
+Pillow
+scikit-learn
+scikit-image
+numpy
+torchsummary