ResNet¶
The ResNet model is based on the “Deep Residual Learning for Image Recognition” paper.
Architecture overview¶
This paper introduces a few tricks to maximize the depth of convolutional architectures that can be trained.
The key takeaways from the paper are the following:
add a shortcut connection in bottleneck blocks to ease the gradient flow
extensive use of batch normalization layers
Model builders¶
The following model builders can be used to instantiate a ResNeXt model, with or
without pre-trained weights. All the model builders internally rely on the
holocron.models.classification.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-50-D from "Bag of Tricks for Image Classification with Convolutional Neural Networks" |
|
ResNet-101 from "Deep Residual Learning for Image Recognition" |
|
ResNet-152 from "Deep Residual Learning for Image Recognition" |