未验证 提交 71330b1d 编写于 作者: C cnn 提交者: GitHub

Update README.md

增加数据来源信息。
上级 5c4b32df
......@@ -2,7 +2,6 @@
=======================
这里将介绍如何在PaddlePaddle下使用AlexNet、VGG、GoogLeNet、ResNet、Inception-v4和Inception-ResNet-v2模型进行图像分类。图像分类问题的描述和这些模型的介绍可以参考[PaddlePaddle book](https://github.com/PaddlePaddle/book/tree/develop/03.image_classification)
所用数据集是[flowers](http://www.robots.ox.ac.uk/~vgg/data/flowers/102/)。因此,CLASS_DIM=102。
## 训练模型
......@@ -32,6 +31,7 @@ paddle.init(use_gpu=False, trainer_count=1)
### 定义参数和输入
设置算法参数(如数据维度、类别数目和batch size等参数),定义数据输入层`image`和类别标签`lbl`
所用数据集是[flowers](http://www.robots.ox.ac.uk/~vgg/data/flowers/102/)。花的种类是102,因此,CLASS_DIM=102。
```python
# Use 3 * 331 * 331 or 3 * 299 * 299 for DATA_DIM in Inception-ResNet-v2.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册