Skip to content

Commit bf4aa3b

Browse files
authored
add wide resnet
1 parent 348a7c3 commit bf4aa3b

File tree

3 files changed

+70
-0
lines changed

3 files changed

+70
-0
lines changed

docs/api/paddle/vision/Overview_cn.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@ paddle.vision 目录是飞桨在视觉领域的高层API。具体如下:
4949
" :ref:`resnet50 <cn_api_paddle_vision_models_resnet50>` ", "50层的ResNet模型"
5050
" :ref:`resnet101 <cn_api_paddle_vision_models_resnet101>` ", "101层的ResNet模型"
5151
" :ref:`resnet152 <cn_api_paddle_vision_models_resnet152>` ", "152层的ResNet模型"
52+
" :ref:`wide_resnet50_2 <_cn_api_paddle_vision_models_wide_resnet50_2>` ", "50层的Wide ResNet模型"
53+
" :ref:`wide_resnet101_2 <_cn_api_paddle_vision_models_wide_resnet101_2>` ", "101层的Wide ResNet模型"
5254
" :ref:`ResNeXt <cn_api_paddle_vision_models_ResNeXt>` ", "ResNeXt模型"
5355
" :ref:`resnext50_32x4d <cn_api_paddle_vision_models_resnext50_32x4d>` ", "ResNeXt-50 32x4d模型"
5456
" :ref:`resnext50_64x4d <cn_api_paddle_vision_models_resnext50_64x4d>` ", "ResNeXt-50 64x4d模型"
Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
.. _cn_api_paddle_vision_models_wide_resnet101_2:
2+
3+
wide_resnet101_2
4+
-------------------------------
5+
6+
.. py:function:: paddle.vision.models.wide_resnet101_2(pretrained=False, **kwargs)
7+
8+
101层的wide_resnet模型,来自论文 `"Wide Residual Networks" <https://arxiv.org/pdf/1605.07146.pdf>`_ 。
9+
10+
参数
11+
:::::::::
12+
- **pretrained** (bool,可选) - 是否加载在imagenet数据集上的预训练权重。默认值:False。
13+
14+
返回
15+
:::::::::
16+
wide_resnet101_2模型,Layer的实例。
17+
18+
代码示例
19+
:::::::::
20+
.. code-block:: python
21+
22+
import paddle
23+
from paddle.vision.models import wide_resnet101_2
24+
25+
# build model
26+
model = wide_resnet101_2()
27+
28+
# build model and load imagenet pretrained weight
29+
# model = wide_resnet101_2(pretrained=True)
30+
31+
x = paddle.rand([1, 3, 224, 224])
32+
out = model(x)
33+
34+
print(out.shape)
Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
.. _cn_api_paddle_vision_models_wide_resnet50_2:
2+
3+
wide_resnet50_2
4+
-------------------------------
5+
6+
.. py:function:: paddle.vision.models.wide_resnet50_2(pretrained=False, **kwargs)
7+
8+
50层的wide_resnet模型,来自论文 `"Wide Residual Networks" <https://arxiv.org/pdf/1605.07146.pdf>`_ 。
9+
10+
参数
11+
:::::::::
12+
- **pretrained** (bool,可选) - 是否加载在imagenet数据集上的预训练权重。默认值:False。
13+
14+
返回
15+
:::::::::
16+
wide_resnet50_2模型,Layer的实例。
17+
18+
代码示例
19+
:::::::::
20+
.. code-block:: python
21+
22+
import paddle
23+
from paddle.vision.models import wide_resnet50_2
24+
25+
# build model
26+
model = wide_resnet50_2()
27+
28+
# build model and load imagenet pretrained weight
29+
# model = wide_resnet50_2(pretrained=True)
30+
31+
x = paddle.rand([1, 3, 224, 224])
32+
out = model(x)
33+
34+
print(out.shape)

0 commit comments

Comments
 (0)