medical-vlm-sam2 / requirements.txt
pascal-maker's picture
Update requirements.txt
1b26d88 verified
raw
history blame contribute delete
567 Bytes
## Core ML and Image Processing
torch>=2.5.1
torchvision>=0.20.1
transformers>=4.30.0
numpy>=1.24.4
pillow>=9.4.0
# Computer Vision and Augmentation
opencv-python>=4.5.0
albumentations>=1.3.0
einops>=0.6.0
# Web Interface
gradio>=4.0.0
# Utilities
tqdm>=4.66.1
hydra-core>=1.3.2
omegaconf>=2.2.0
pyyaml>=6.0
packaging>=21.0
# Additional dependencies that might be needed
iopath>=0.1.10
portalocker>=2.0.0
# Optional but recommended
matplotlib>=3.5.0
seaborn>=0.11.0
scikit-image>=0.19.0
scikit-learn>=1.0.0
# For better performance (optional)
accelerate>=0.20.0