Skip to content

TanmayBansode/seds

Repository files navigation

SEDS - Smart Eye Detection System

A YOLOv8-based object detection system that supports inference on images, videos, and live camera feeds with optional server integration for real-time alerts.

Features

  • 🎯 Object Detection - Detect objects using custom-trained YOLO models
  • 📊 Congestion Analysis - Calculate real-time congestion levels
  • 🎥 Multi-source Input - Process images, videos, folders, or live camera feeds
  • 🖥️ Server Integration - Send detection alerts to a backend server

Project Structure

seds/
├── inference.py        # Main CLI script for batch inference
├── client.py           # Real-time detection client with server integration
├── congestion.py       # Congestion analysis module
├── yolo.py             # Core YOLO utility functions
├── config.py           # Model configuration settings
├── model_info.py       # Utility to inspect model details
├── requirements.txt    # Python dependencies
├── your_model.pt       # Your trained YOLO model
├── test_images/        # Sample test images
└── output/             # Inference results output directory

Installation

Prerequisites

  • Python 3.8+
  • CUDA-compatible GPU (optional, for faster inference)

Setup

# Clone the repository
git clone https://github.com/TanmayBansode/seds.git
cd seds

# Create virtual environment (recommended)
python -m venv venv
venv\Scripts\activate  # Windows
# source venv/bin/activate  # Linux/macOS

# Install dependencies
pip install -r requirements.txt

Usage

1. Configure Your Model

Edit config.py to specify your trained model:

model_file = "your_model.pt"

2. Run Inference (inference.py)

# Single image
python inference.py -i path/to/image.jpg

# Video
python inference.py -v path/to/video.mp4

# Folder of images/videos
python inference.py -f path/to/folder/

Results are saved to output/ with labeled media and detection details in .txt files.

3. Real-time Detection (client.py)

python client.py

Processes live camera feed with real-time detection and sends alerts to a configured server.

4. Congestion Analysis (congestion.py)

python congestion.py --source path/to/video.mp4

Calculates congestion levels based on detected object coverage.

5. Model Info (model_info.py)

python model_info.py

Displays model architecture, class names, and hyperparameters.


File Descriptions

File Description
inference.py CLI tool for batch inference on images, videos, or folders
client.py Real-time detection client with server integration
congestion.py Congestion level calculation and monitoring
yolo.py Core YOLO functions: model loading, inference utilities
config.py Configuration file specifying which model to use
model_info.py Utility to inspect model details

Supported Formats

Images: jpg, jpeg, png, bmp, webp, tif, tiff

Videos: mp4, avi, mov, mkv, webm

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages