🚀 Daily Object Detection Pipeline: YOLOv5 + AWS

An automated pipeline that performs daily object detection on Columbus Circle EarthCam images using YOLOv5, deployed with AWS Lambda for seamless cloud processing and visualization.

⚡️ View the GitHub Page to Interact with Plots and Visualizations ⚡️

This project performs daily object detection on EarthCam images from Columbus Circle, NY, analyzing and visualizing counts of people, cars, buses, and more using YOLOv5. The system is deployed with AWS services, making it cost-efficient and fully automated.

🌟 Key Features

• Web Scraping: Selenium with Headless Chrome
• Object Detection: Ultralytics YOLOv5
• Data Visualization: Plotly
• AWS Deployment: Docker, Lambda, ECR, EventBridge, S3

📊 Results

Two interactive data visualizations are updated daily around 5 PM ET:

• 🖼️ Daily YOLOv5 Analysis

• 📈 Object Counts Over Time: Explore this plot to observe the trends throughout the COVID-19 pandemic and beyond!

⚙️ AWS Deployment Architecture

Workflow Overview:

1. Local Development:

   • Develop and test locally, building a Docker image using the Dockerfile.
   • Test locally with the AWS Lambda Runtime Interface Emulator to fine-tune memory and runtime settings.

2. Docker Image Deployment:

   • Push the Docker image to an AWS Elastic Container Registry (ECR).

3. Lambda Function:

   • Create an AWS Lambda function linked to the Docker image in ECR. The Lambda function handles object detection and updates the data daily.

4. S3 Buckets:

   • Model & Data: One bucket stores YOLOv5 model weights and the results database.
   • Static Websites: Two buckets serve as static websites for interactive Plotly visualizations, updated daily by the Lambda function.

5. Automation & Monitoring:

   • EventBridge: Triggers the Lambda function daily using a cron expression.
   • CloudWatch: Captures logs and errors for easy debugging.

💡 Tip: If you encounter S3 access issues, check your Lambda function’s IAM Role and S3 bucket policies.

Why AWS Lambda?

AWS Lambda is ideal for this project due to its serverless, event-driven architecture. The app runs for just a few seconds daily, making it highly cost-effective — often falling within the free tier or costing just a few cents per month.

---

Feel free to explore my projects and reach out for collaborations or just to chat about data science and AI!