Welcome to our project where we use YOLOv5 to detect drones in images and videos, and also provide a GUI application to make it more accessible for everyone.
To use our drone detection app, follow these steps:
-
Clone the repository:
git clone https://github.com/your_username/drone-detection.git
-
Install the required packages:
pip install -r requirements.txt
-
Download the
Drone_Detection.pt
model file from the release page.
-
To run the drone detection model in the Jupyter notebook, open
Drone_detection.ipynb
and run the cells. -
To run the GUI application, run
GUI_application.py
using the following command:python GUI_application.py
-
The application window will appear where you can select an image or video file to detect drones in. The output will be shown on the screen. (You may need to change path of pytorch model in GUI_application.py)
-
We also have dataset hosted on here.
This project was made possible with the contributions of:
- Siddheshpatil
- Siddhi Patil
Thank you for your hard work and dedication to this project.