Skip to content

rlpddejong/SurgeNetDINO

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🧠 SurgeNetDINO: Towards Effective Surgical Representation Learning with DINO Models

This repository contains the official code and pretrained model weights for the paper:

“Towards Effective Surgical Representation Learning with DINO Models”
(Accepted for Medical Imaging with Deep Learning (MIDL) 2026)

Find the paper here!

⚙️ Requirements

  • Python 3.9+
  • torch
  • timm

🐍 Installation example using Conda

# Create a new conda environment with Python 3.9
conda create -n SurgeNetDINO python=3.9 -y

# Activate the environment
conda activate SurgeNetDINO

# Install PyTorch and timm
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
pip install timm

💾 Loading models

After installations, you can use load_weights.py to load the pretrained DINO models.

⬇️ Download the model weights here

Alternatively, you can download the model weights using the provided links below.

Model Variant Download
DINOv1 ViT-s Download
DINOv1 ViT-b Download
DINOv2 ViT-s Download
DINOv2 ViT-b Download
DINOv2 ViT-l Download
DINOv3 ViT-s Download
DINOv3 ViT-b Download
DINOv3 ViT-l Download

📄 Pretraining configurations

The pretraining configurations can be found in dinov1_configs, dinov2_configs, and dinov3_configs. Note that for dinov1, we provide a .sh file instead of a configuration file following the original implementation. For further pretraining instructions, please refer to the original implementations of DINO: DINOv1, DINOv2, and DINOv3.

📚 Citation

If you use these models or the dataset in your work, please cite our paper:

@inproceedings{
  jong2026towards,
  title={Towards Effective Surgical Representation Learning with {DINO} Models},
  author={Ronald L.P.D. de Jong and Yiping Li and Tim J. M. Jaspers and Romy C. van Jaarsveld and Gino M. Kuiper and Franco Badaloni and Richard van Hillegersberg and Jelle P. Ruurda and Fons van der Sommen and Josien P.W. Pluim and Marcel Breeuwer},
  booktitle={Medical Imaging with Deep Learning},
  year={2026},
  url={https://openreview.net/forum?id=6FoIDPKzRV}
}

📬 Contact

For questions or issues regarding this repository, please contact the corresponding author:

Ronald L.P.D. de Jong
Email: r.l.p.d.d.jong@tue.nl

⚖️ License

  • Code: MIT — see LICENSE (permissive; commercial use permitted).
  • Pretrained model weights: CC-BY-NC-SA — non-commercial share-alike. The weights and any derivative models that include these weights are NOT cleared for commercial use. See LICENSE_MODELS for details and the precise license text.

🙏 Acknowledgements

We would like to thank the authors and maintainers of SurgeNetXL and the original DINO repositories for making their work publicly available:

Their open-source contributions provided the foundation for our work on surgical representation learning.

About

[MIDL 2026] - Official repo for the paper: "SurgeNetDINO: Towards Effective Surgical Representation Learning with DINO Models""

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors