Shortcuts

mmcls.models.LeNet5

class mmcls.models.LeNet5(num_classes=- 1)[source]

LeNet5 backbone.

The input for LeNet-5 is a 32×32 grayscale 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.