search

mazurowski-lab / finetune-SAM

This is an official repo for fine-tuning SAM to customized medical images.
https://arxiv.org/abs/2404.09957
Apache License 2.0
136 stars 22 forks source link

Self-supervised training #6

Closed ShijianRuan closed 5 months ago

ShijianRuan commented 6 months ago

Is there code for self-supervised training? If not, can you provide reference code?

haoyudong-97 commented 6 months ago

Most of the self-supervised learning codes are inherited from the official MAE repo (https://github.com/facebookresearch/mae/tree/main). We adjust the input size to 1024 to fit the SAM's requirement.