Quellcode durchsuchen

新增依赖文件

liyan vor 7 Monaten
Ursprung
Commit
7c72940a9b
2 geänderte Dateien mit 24 neuen und 0 gelöschten Zeilen
  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