Browse Source

新增依赖文件

liyan 7 months ago
parent
commit
7c72940a9b
2 changed files with 24 additions and 0 deletions
  1. 20 0
      README.md
  2. 4 0
      requirements.txt

+ 20 - 0
README.md

@@ -15,6 +15,26 @@ classification-models-tensorflow
     └── train_vgg16.py  # VGG16模型训练脚本
 ```
 
+## 运行环境
+```text
+python=3.9
+pillow
+scipy
+numpy==1.24.4
+tensorflow==2.10.0
+```
+
+## 环境搭建
+- 构建虚拟环境,如果存在,请忽略
+```shell
+conda create -n tensorflow python=3.9
+conda activate tensorflow
+```
+- 安装依赖
+```shell
+pip install -r requirements.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
+```
+
 ## 训练命令
 - AlexNet
 ```shell

+ 4 - 0
requirements.txt

@@ -0,0 +1,4 @@
+pillow
+scipy
+numpy==1.24.4
+tensorflow==2.10.0