Projects Details

Project Name: Train Wagon Detection System

Objective: Detect and identify train wagons from images or video streams using computer vision and deep learning.

Model Architecture:

Image Preprocessing: Prepares railway images and video frames by resizing, normalizing, and enhancing the input data for reliable wagon detection.
Feature Extraction: Extracts important visual features such as wagon shapes, edges, dimensions, and other patterns from railway images.
Object Detection Model: Uses a deep learning object-detection model such as YOLO to detect and localize train wagons within images or video frames.
Classification & Tracking: Identifies detected wagons and tracks their movement across consecutive video frames for consistent monitoring.
Tools: Python, YOLO, OpenCV, PyTorch, Computer Vision.