smartrail-obstacle-detector 0.1.1

Railway Obstacle Detection Model for SmartRailShield using YOLOv8

SmartRail Obstacle Detector

A professional-grade railway obstacle detection system powered by YOLOv8. Part of the SmartRailShield safety suite.

Features

• Real-time obstacle detection using calibrated YOLOv8 models.
• Automatic GPS-based location approximation for detected objects.
• Easy integration with Flask, Streamlit, or other Python backends.
• Pre-trained on railway-specific obstacles (trains, people, animals, vehicles on tracks).

Installation

pip install smartrail-obstacle-detector

Quick Start

from railway_obstacle_detector import ObstacleDetector

# Initialize detector
detector = ObstacleDetector()

# Detect obstacles
obstacles, _ = detector.get_obstacles(current_location={"lat": 13.0827, "lon": 80.2707})

for obs in obstacles:
    print(f"Detected {obs['class']} at {obs['lat']}, {obs['lon']}")

# Cleanup
detector.release()

License

MIT