resnet50¶
- paddle.vision.models. resnet50 ( pretrained=False, **kwargs ) [source]
-
ResNet 50-layer model from “Deep Residual Learning for Image Recognition”.
- Parameters
-
pretrained (bool, optional) – Whether to load pre-trained weights. If True, returns a model pre-trained on ImageNet. Default: False.
**kwargs (optional) – Additional keyword arguments. For details, please refer to ResNet.
- Returns
-
Layer. An instance of ResNet 50-layer model.
Examples
>>> import paddle >>> from paddle.vision.models import resnet50 >>> # build model >>> model = resnet50() >>> # build model and load imagenet pretrained weight >>> # model = resnet50(pretrained=True) >>> x = paddle.rand([1, 3, 224, 224]) >>> out = model(x) >>> print(out.shape) [1, 1000]