arm-model / model /requirements.txt
pragadeeshv23's picture
Upload folder using huggingface_hub
5b86813 verified
# ═══════════════════════════════════════════════════════════════════════════
# Road Anomaly Detection – YOLOv11 + CNN-BiGRU
# Based on: Nature Scientific Reports (Nov 2025)
# ═══════════════════════════════════════════════════════════════════════════
# --- Core ---
ultralytics>=8.3.0 # YOLOv11 support
torch>=2.1.0
torchvision>=0.16.0
torchaudio>=2.1.0
# --- Computer Vision ---
opencv-python>=4.8.0
numpy>=1.24.0
pillow>=10.0.0
# --- Data & ML ---
scikit-learn>=1.3.0
scipy>=1.11.0
pandas>=2.1.0
pyyaml>=6.0
# --- Visualisation & Logging ---
matplotlib>=3.7.0
tensorboard>=2.14.0
tqdm>=4.66.0
# --- Export / Deployment ---
onnx>=1.15.0
onnxruntime>=1.16.0
# tensorrt # Install separately: pip install tensorrt
# tensorflow # For TFLite export (optional)
# --- Optional: GAN augmentation ---
# torchgan # If using GAN-based augmentation