coronary-syntax-prediction / requirements.txt
MesserMMP's picture
Add model code and full model weights
f621d73
Raw
History Blame Contribute Delete
367 Bytes
# Core ML/DL
torch>=2.0.0
torchvision>=0.15.0
lightning>=2.1.0
pytorchvideo @ git+https://github.com/facebookresearch/pytorchvideo@main#egg=pytorchvideo
# Data processing
numpy>=1.24.0
scikit-learn>=1.3.0
tqdm>=4.65.0
pydicom>=2.4.0
python-gdcm>=3.0.10
# Visualization
plotly>=5.17.0
scipy>=1.11.0
tensorboard>=2.9
tensorboardX>=2.6
# CLI & Utilities
click>=8.1.0