ProFound / requirements.txt
Anonymise's picture
Update requirements.txt
385ada6 verified
raw
history blame contribute delete
788 Bytes
# Deep Learning and AI
torch
torchvision
--extra-index-url https://download.pytorch.org/whl/cu118
# Gradio for Web Interface
gradio==5.35.0
spaces==0.36.0
# Hugging Face Hub for Model/Data Loading
huggingface_hub
safetensors==0.5.3
# Medical Image Processing
nibabel==5.3.2
SimpleITK==2.5.0
scikit-image==0.25.2
scikit-learn==1.7.0
scipy==1.15.3
# MONAI for Medical Image Analysis
monai==1.4.0
# Basic Scientific Computing
numpy==1.26.4
matplotlib==3.10.1
pillow==10.4.0
# Data Processing
pandas==2.2.3
# Logging and Utilities
tqdm==4.67.1
# Additional dependencies for model architecture
einops==0.8.1
timm==1.0.15
prettytable==3.16.0
surface-distance==0.1
torchmetrics==1.7.3
# Advanced Visualization
plotly==5.18.0
opencv-python==4.10.0.84
# Captum for Grad-CAM
captum==0.8.0