Base Classification Model

class vformer.common.base_model.BaseClassificationModel(img_size, patch_size, in_channels=3, pool='cls')[source]
Parameters

  • img_size (int) – Size of the image

  • patch_size (int or tuple(int)) – Size of the patch

  • in_channels (int) – Number of channels in input image

  • pool (str) – Feature pooling type, must be one of {mean, cls}