starry / backend /python-services /requirements.txt
k-l-lambda's picture
feat: add Python ML services (CPU mode) with model download
2b7aae2
# Core dependencies
numpy>=1.21.0
opencv-python>=4.5.0
Pillow>=8.0.0
PyYAML>=5.4.0
# Communication
pyzmq>=22.0.0
msgpack>=1.0.0
# PyTorch (for layout, mask, semantic, gauge, loc services)
torch>=1.9.0
torchvision>=0.10.0
# TensorFlow (for ocr, brackets services)
# Note: TensorFlow 2.x required for Keras SavedModel support
tensorflow>=2.6.0
# Optional: GPU support
# Install with: pip install tensorflow[and-cuda] for TF GPU support
# PyTorch GPU support comes with the default torch installation from pytorch.org