- class mmpretrain.models.heads.MAEPretrainHead(loss, norm_pix=False, patch_size=16, in_channels=3)¶
Head for MAE Pre-training.
Construct the reconstruction target.
In addition to splitting images into tokens, this module will also normalize the image according to
- loss(pred, target, mask)¶
Split images into non-overlapped patches.