MCPcopy Index your code
hub / github.com/YtongXie/UniMiSS-code

github.com/YtongXie/UniMiSS-code @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
1,146 symbols 2,975 edges 88 files 214 documented · 19%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

UniMiSS & UniMiSS+

This is the official pytorch implementation of our ECCV 2022 paper "UniMiSS: Universal Medical Self-Supervised Learning via Breaking Dimensionality Barrier" and extended IEEE-TPAMI paper "UniMiSS+: Universal Medical Self-Supervised Learning From Cross-Dimensional Unpaired Data".

To do

  • [x] UniMiSS+ fine-tuning code
  • [x] UniMiSS+ pre-training code and weights
  • [x] UniMiSS fine-tuning code
  • [x] UniMiSS pre-training code and weights

Citation

If this code is helpful for your study, please cite:

@article{UniMiSS,
  title={UniMiSS: Universal Medical Self-Supervised Learning via Breaking Dimensionality Barrier},
  author={Xie, Yutong and Zhang, Jianpeng and Xia, Yong and Wu, Qi},
  booktitle={ECCV},
  year={2022}
}
@article{UniMiSS+,
  title={UniMiSS+: Universal Medical Self-Supervised Learning From Cross-Dimensional Unpaired Data},
  author={Xie, Yutong and Zhang, Jianpeng and Xia, Yong and Wu, Qi},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
  year={2024},
  publisher={IEEE}
}

Acknowledgements

Part of codes is reused from the DINO. Thanks to Caron et al. for the codes of DINO.

Contact

Yutong Xie (yutong.xie678@gmail.com)

Core symbols most depended-on inside this repo

print
called by 179
UniMiSS/Downstream/BCV/MiTnnu/network_architecture /utils.py
print
called by 151
UniMiSSPlus/Downstream/3D/BCV/MiTnnu/network_architecture/utils.py
print
called by 75
UniMiSSPlus/Downstream/2D/Cls/net/utils.py
print_to_log_file
called by 48
UniMiSS/Downstream/BCV/MiTnnu/training/network_training /network_trainer.py
print_to_log_file
called by 48
UniMiSSPlus/Downstream/3D/BCV/MiTnnu/training/network_training/network_trainer.py
print
called by 45
UniMiSSPlus/Downstream/2D/Seg/net/utils.py
print
called by 42
UniMiSSPlus/utils.py
trunc_normal_
called by 32
UniMiSS/Downstream/RICORD/nets/utils.py

Shape

Method 677
Function 296
Class 173

Languages

Python100%

Modules by API surface

UniMiSSPlus/utils.py62 symbols
UniMiSSPlus/MiTplus.py55 symbols
UniMiSSPlus/Downstream/2D/Seg/net/utils.py54 symbols
UniMiSSPlus/Downstream/2D/Cls/net/utils.py54 symbols
UniMiSSPlus/Downstream/3D/BCV/MiTnnu/network_architecture/utils.py47 symbols
UniMiSS/utils.py47 symbols
UniMiSS/Downstream/BCV/MiTnnu/network_architecture /utils.py47 symbols
UniMiSSPlus/Downstream/3D/BCV/MiTnnu/network_architecture/MiTPlus.py38 symbols
UniMiSS/Downstream/BCV/MiTnnu/network_architecture /MiT_encoder.py34 symbols
UniMiSS/models/MiT.py33 symbols
UniMiSS/Downstream/BCV/MiTnnu/network_architecture /MiT.py32 symbols
UniMiSS/models/MiT_encoder.py29 symbols

For agents

$ claude mcp add UniMiSS-code \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact