提交 e07a2111 编写于 作者: littletomatodonkey's avatar littletomatodonkey

add play around

上级 b11cd64c
...@@ -3,21 +3,21 @@ ...@@ -3,21 +3,21 @@
--- ---
## 1.简介 ## 1.简介
本文档介绍ImageNet1k和Flower102数据准备过程。 本文档介绍ImageNet1k和flowers102数据准备过程。
以及PaddleClas提供了丰富的[预训练模型](../models/models_intro.md) 以及PaddleClas提供了丰富的[预训练模型](../models/models_intro.md)
## 2.数据集准备 ## 2.数据集准备
数据集 | 训练集大小 | 测试集大小 | 类别数 | 备注| 数据集 | 训练集大小 | 测试集大小 | 类别数 | 备注|
:------:|:---------------:|:---------------------:|:-----------:|:-----------: :------:|:---------------:|:---------------------:|:-----------:|:-----------:
[Flower102](https://www.robots.ox.ac.uk/~vgg/data/flowers/102/)|1k | 6k | 102 | [flowers102](https://www.robots.ox.ac.uk/~vgg/data/flowers/102/)|1k | 6k | 102 |
[ImageNet1k](http://www.image-net.org/challenges/LSVRC/2012/)|1.2M| 50k | 1000 | [ImageNet1k](http://www.image-net.org/challenges/LSVRC/2012/)|1.2M| 50k | 1000 |
数据格式 * 数据格式
按照如下结构组织数据,其中train_list.txt 和val_list.txt的格式形如 按照如下结构组织数据,其中train_list.txt 和val_list.txt的格式形如
``` ```shell
#每一行采用"空格"分隔图像路径与标注 # 每一行采用"空格"分隔图像路径与标注
ILSVRC2012_val_00000001.JPEG 65 ILSVRC2012_val_00000001.JPEG 65
... ...
...@@ -44,26 +44,26 @@ PaddleClas/dataset/imagenet/ ...@@ -44,26 +44,26 @@ PaddleClas/dataset/imagenet/
|_ train_list.txt |_ train_list.txt
|_ val_list.txt |_ val_list.txt
``` ```
### Flower ### Flowers102
[VGG官方网站](https://www.robots.ox.ac.uk/~vgg/data/flowers/102/)下载后的数据,解压后包括 [VGG官方网站](https://www.robots.ox.ac.uk/~vgg/data/flowers/102/)下载后的数据,解压后包括
jpg/ jpg/
setid.mat setid.mat
imagelabels.mat imagelabels.mat
将以上文件放置在PaddleClas/dataset/flower102/下 将以上文件放置在PaddleClas/dataset/flowers102/下
通过运行generate_flower_list.py生成train_list.txt和val_list.txt 通过运行generate_flowers102_list.py生成train_list.txt和val_list.txt
```bash ```bash
python generate_flower_list.py jpg train > train_list.txt python generate_flowers102_list.py jpg train > train_list.txt
python generate_flower_list.py jpg valid > val_list.txt python generate_flowers102_list.py jpg valid > val_list.txt
``` ```
按照如下结构组织数据: 按照如下结构组织数据:
```bash ```bash
PaddleClas/dataset/flower102/ PaddleClas/dataset/flowers102/
|_ jpg/ |_ jpg/
| |_ image_03601.jpg | |_ image_03601.jpg
| |_ ... | |_ ...
| |_ image_02355.jpg | |_ image_02355.jpg
|_ train_list.txt |_ train_list.txt
......
...@@ -12,7 +12,6 @@ ...@@ -12,7 +12,6 @@
#See the License for the specific language governing permissions and #See the License for the specific language governing permissions and
#limitations under the License. #limitations under the License.
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
__all__ = ['CELoss', 'MixCELoss', 'GoogLeNetLoss', 'JSDivLoss'] __all__ = ['CELoss', 'MixCELoss', 'GoogLeNetLoss', 'JSDivLoss']
...@@ -26,7 +25,7 @@ class Loss(object): ...@@ -26,7 +25,7 @@ class Loss(object):
def __init__(self, class_dim=1000, epsilon=None): def __init__(self, class_dim=1000, epsilon=None):
assert class_dim > 1, "class_dim=%d is not larger than 1" % (class_dim) assert class_dim > 1, "class_dim=%d is not larger than 1" % (class_dim)
self._class_dim = class_dim self._class_dim = class_dim
if epsilon and epsilon >= 0.0 and epsilon <= 1.0: if epsilon is not None and epsilon >= 0.0 and epsilon <= 1.0:
self._epsilon = epsilon self._epsilon = epsilon
self._label_smoothing = True self._label_smoothing = True
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册