Shape2Force / requirements.txt
kaveh's picture
added
2b9ff22
# Shape2Force (S2F) - GUI and training
torch>=2.0.0
torchvision>=0.15.0
numpy>=1.20.0
opencv-python>=4.5.0
streamlit>=1.28.0
matplotlib>=3.5.0
Pillow>=9.0.0
plotly>=5.14.0
# Training & evaluation
torchmetrics>=1.0.0
diffusers>=0.20.0
tqdm>=4.65.0
scikit-image>=0.19.0
scipy>=1.9.0
scikit-learn>=1.0.0
pandas>=1.3.0