Prostate-Inference / requirements.txt
Anirudh Balaraman
add CD
30f1102
raw
history blame contribute delete
917 Bytes
# ---- Core Python ----
numpy==1.26.4
scipy==1.10.0
pandas==2.2.3
PyYAML
# ---- PyTorch & ecosystem ----
torch==2.5.1
torchvision==0.20.1
torcheval==0.0.7
torchmetrics==1.6.0
pytorch-lightning==2.4.0
triton==3.1.0
# ---- MONAI / medical imaging ----
monai==1.4.0
itk>=5.3.0
SimpleITK==2.4.0
pynrrd==1.1.1
nibabel==5.3.2
scikit-image==0.24.0
scikit-learn==1.5.2
opencv-python-headless==4.10.0.84
picai_prep==2.1.13
# ---- Transformers / Hugging Face ----
huggingface_hub==1.3.3
# ---- Visualization ----
matplotlib==3.9.3
seaborn==0.13.2
plotly
# ---- Experiment tracking ----
wandb==0.16.6
tensorboard==2.18.0
# ---- Utilities ----
tqdm==4.67.1
gdown==5.2.0
requests
filelock
packaging
# ---- Streamlit ----
streamlit==1.50.0
# ---- Grad-CAM ----
grad-cam @ git+https://github.com/jacobgil/pytorch-grad-cam.git@781dbc0d16ffa95b6d18b96b7b829840a82d93d1
# --- Development Tools ---
ruff
mypy
pytest
pytest-mock