瀏覽代碼

新增依赖文件

liyan 7 月之前
父節點
當前提交
7c72940a9b
共有 2 個文件被更改,包括 24 次插入0 次删除
  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