miccaiif / DGMIL

Official PyTorch implementation of our MICCAI 2022 paper: DGMIL: Distribution Guided Multiple Instance Learning for Whole Slide Image Classification.
32 stars 3 forks source link

The problem of model.py #9

Closed YuRongHao2233 closed 10 months ago

YuRongHao2233 commented 1 year ago

I didn't find the model of mae,and the function of mae is a projection,the same as projection768

YuRongHao2233 commented 1 year ago

@miccaiif

miccaiif commented 10 months ago

@miccaiif

Thank you for your attention! We pre-trained a feature extractor using the official MAE code and extracted features based on it. The official code of MAE can be referred to at the link https://github.com/facebookresearch/mae. During our training process, we only use a projector for feature mapping, while the features pre-extracted by MAE remain unchanged. This also means that the MAE model is fixed in our experiments.