Base Classification Model

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

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: {“mean”,”cls”}

Feature pooling type