AlexNet¶
- class mmpretrain.models.backbones.AlexNet(num_classes=-1)[source]¶
AlexNet backbone.
The input for AlexNet is a 224x224 RGB image.
- Parameters:
num_classes (int) – number of classes for classification. The default value is -1, which uses the backbone as a feature extractor without the top classifier.