|
@@ -15,6 +15,26 @@ classification-models-tensorflow
|
|
└── train_vgg16.py # VGG16模型训练脚本
|
|
└── 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
|
|
- AlexNet
|
|
```shell
|
|
```shell
|