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