@@ -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
+- 安装依赖
+pip install -r requirements.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
## 训练命令
- AlexNet
```shell
@@ -0,0 +1,4 @@