ResNet¶
The ResNet model is based on the Deep Residual Learning for Image Recognition paper.
Note
The bottleneck of TorchVision places the stride for downsampling to the second 3x3 convolution while the original paper places it to the first 1x1 convolution. This variant improves the accuracy and is known as ResNet V1.5.
Model builders¶
The following model builders can be used to instantiate a ResNet model, with or
without pre-trained weights. All the model builders internally rely on the
torchvision.models.resnet.ResNet
base class. Please refer to the source
code for
more details about this class.
|
ResNet-18 from Deep Residual Learning for Image Recognition. |
|
ResNet-34 from Deep Residual Learning for Image Recognition. |
|
ResNet-50 from Deep Residual Learning for Image Recognition. |
|
ResNet-101 from Deep Residual Learning for Image Recognition. |
|
ResNet-152 from Deep Residual Learning for Image Recognition. |