rklb_materials / requirements.txt
gphua1's picture
Auto-download model from HuggingFace Hub to bypass git size limits
5607298
raw
history blame contribute delete
543 Bytes
# Core ML Framework
torch>=2.1.0
torchvision>=0.16.0
timm>=0.9.12
# Data Processing
numpy>=1.24.0
opencv-python-headless>=4.8.0 # Use headless for servers
Pillow>=10.0.0
albumentations>=1.3.0
# ML & Metrics
scikit-learn>=1.3.0
# Web Framework (choose one)
streamlit>=1.28.0 # For web UI
fastapi>=0.104.0 # For API server
uvicorn[standard]>=0.24.0 # ASGI server
# Utilities
tqdm>=4.65.0
pydantic>=2.5.0
huggingface-hub>=0.19.0
# Optional (uncomment if needed)
# wandb>=0.15.0 # Experiment tracking
# matplotlib>=3.7.0 # Visualization