A repository for YOLO-based computer vision models for object detection.
Clone this repository to your local machine:
git clone https://github.com/YOUR_USERNAME/Vision-YOLO-Models.git
cd Vision-YOLO-ModelsUsing a virtual environment helps isolate dependencies and avoid conflicts.
For Windows
python -m venv .venv
.venv\Scripts\activateFor MacOS / Linux
python -m venv .venv
source .venv/bin/activateInstall all necessary packages from requirements.txt:
pip install -r requirements.txtEnsure all dependencies were installed correctly:
pip listTo run the YOLO model on an image file:
python vision.py model.pt input.jpg --show TrueRunning on a Video File
python vision.py model.pt video.mp4 --show TrueRunning on a Live Camera
python vision.py 0 --show TrueRunning on Image and saving to disk
python vision.py model.pt image.jpg --destination .\output\Running on Image and showing to screen
python vision.py model.pt image.jpg --show TrueRunning on Video, showing to screen, and Using GPU (By default, GPU use is True)
python vision.py model.pt image.jpg --show True --gpu TrueIf new dependencies are added, update requirements.txt:
pip freeze > requirements.txtWe welcome contributions! Follow these steps:
Fork this repository 🍴 Create a new branch (git checkout -b feature-branch) Make your changes and commit (git commit -m "Add new feature") Push to your branch (git push origin feature-branch) Open a Pull Request for review! ✅
This project is licensed under the MIT License. Feel free to use and modify it as needed.
🔥 Happy Coding! 🚀 If you find this project useful, don’t forget to star ⭐ the repository!