ResNet50

Classes

class backbone.ResNet50.Bottleneck(inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm_layer=None)

Bases: Module

expansion: int = 4

forward(x)

Return type:

Tensor

class backbone.ResNet50.ResNet(block, layers, num_classes=1000, zero_init_residual=False, groups=1, pretrained=False, width_per_group=64, replace_stride_with_dilation=None, norm_layer=None)

Bases: MammothBackbone

forward(x, returnt='out')

Return type:

Tensor

set_grad_filter(filter_s, enable)

set_return_prerelu(enable=True)

to(device, **kwargs)

Functions

backbone.ResNet50.conv1x1(in_planes, out_planes, stride=1)

1x1 convolution

Return type:

Conv2d

backbone.ResNet50.conv3x3(in_planes, out_planes, stride=1, groups=1, dilation=1)

3x3 convolution with padding

Return type:

Conv2d

backbone.ResNet50.resnet50(num_classes, pretrained=False, **kwargs)

ResNet-50 model from “Deep Residual Learning for Image Recognition”.

Parameters:

• pretrained (bool) – If True, returns a model pre-trained on ImageNet

• progress (bool) – If True, displays a progress bar of the download to stderr

Return type:

ResNet