Object Detection with PyTorch & OpenCV

Overview

This project demonstrates object detection using PyTorch and OpenCV. It allows detection on live video (webcam), image files, and video files.

Installation

Clone the Repository

git clone https://github.com/camelbyte/object-detection.git
cd object-detection

Install Dependencies

Don't forget to install requirements.

pip install -r requirements.txt

Usage

Run Object Detection

On Default Source (Webcam)

python object_detection.py

On a Video File

python object_detection.py --source video.mp4

On an Image File

python object_detection.py --source image.jpg

Requirements

• Python 3.x
• PyTorch
• OpenCV
• Other dependencies listed in requirements.txt

License

This project is open-source. Feel free to modify and use it as needed.

---

Happy coding! 🚀