search

chinmay5 / vit_ae_plus_plus

Code base for the paper ViT-AE++: Improving Vision Transformer Autoencoder for Self-supervised Medical Image Representations
Creative Commons Zero v1.0 Universal
40 stars 4 forks source link

Pretrained weights #4

Open mahaling opened 1 year ago

mahaling commented 1 year ago

Can you please share some details about the pretrained weights you have posted. Is this something we can use to extract embeddings on a new dataset?. If not, can you please share the model weights for the datasets you have trained on. Thanks

chinmay5 commented 1 year ago

We will upload pretrained weights soon. In our experiments, the model weights are dataset-specific. It would be difficult for the model to work on a different dataset unless the new dataset is very similar to the Brats and EGD datasets.

27yw commented 8 months ago

hello, this page seems to be crashed can u pls share the checkpoint again?thank u!