
multi_task_test:利用tensorflow.contrib.slim训练适用于多任务的简易CNN分类模型
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
multi_task_test项目采用TensorFlow的slim库来构建和训练一个简单的卷积神经网络(CNN)模型,旨在实现多种相关任务的高效联合学习。
使用tensorflow.contrib.slim训练用于多任务的简单CNN分类模型的方法如下:
1. 克隆该项目。
2. 创建目录`./datasets/images/`
3. 生成训练图像:运行命令 `python3 generate_train_data.py`
4. 生成tfrecord文件,执行以下命令:
```
python3 generate_tfrecord.py --images_path ./datasets/images/ --output_path ./datasets/train.record
```
5. 创建目录`./training/`
6. 训练CNN模型:运行命令 `python3 train.py --record_path ./datasets/train.record --logdir ./tra`
全部评论 (0)
还没有任何评论哟~


