requirements.txt 401 B

1234567891011121314151617181920212223242526
  1. # Core training stack
  2. torch>=2.2
  3. torchvision>=0.17
  4. numpy>=1.24
  5. Pillow>=10.0
  6. PyYAML>=6.0
  7. # Medical segmentation losses and metrics
  8. monai>=1.3
  9. # XNet2d wavelet and VMamba dependencies
  10. ptwt>=0.1.9
  11. PyWavelets>=1.5
  12. timm>=1.0
  13. fvcore>=0.1.5
  14. einops>=0.7
  15. packaging>=23.0
  16. triton>=2.2; platform_system == "Linux"
  17. # Experiment logging and utilities
  18. swanlab>=0.6
  19. tqdm>=4.66
  20. matplotlib>=3.8
  21. # Tests
  22. pytest>=8.0