https://github.com/rizwanmunawar/yolov7-pose-estimation
YOLOv7 Pose estimation using OpenCV, PyTorch
https://github.com/rizwanmunawar/yolov7-pose-estimation
medium-article opencv opencv-python pose-estimation yolov7
Last synced: 16 days ago
JSON representation
YOLOv7 Pose estimation using OpenCV, PyTorch
- Host: GitHub
- URL: https://github.com/rizwanmunawar/yolov7-pose-estimation
- Owner: RizwanMunawar
- License: gpl-3.0
- Created: 2022-08-16T05:29:26.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2025-02-09T12:05:20.000Z (2 months ago)
- Last Synced: 2025-04-01T13:08:47.430Z (23 days ago)
- Topics: medium-article, opencv, opencv-python, pose-estimation, yolov7
- Language: Python
- Homepage:
- Size: 3.65 MB
- Stars: 363
- Watchers: 5
- Forks: 84
- Open Issues: 10
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE
Awesome Lists containing this project
README
# yolov7-pose-estimation
### Steps to Run Code
- **Google Colab Users**: First, mount the drive:
```python
from google.colab import drive
drive.mount("/content/drive")
```
- **Clone the repository**:
```bash
git clone https://github.com/RizwanMunawar/yolov7-pose-estimation.git
```
- **Go to the cloned folder**:
```bash
cd yolov7-pose-estimation
```
- **Create a virtual environment** (recommended):
```bash
# Linux
python3 -m venv psestenv
source psestenv/bin/activate
# Windows
python3 -m venv psestenv
cd psestenv/Scripts
activate
```
- **Upgrade pip**:
```bash
pip install --upgrade pip
```
- **Install requirements**:
```bash
pip install -r requirements.txt
```
- **Download YOLOv7 weights** and move to the working directory:
[yolov7-w6-pose.pt](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6-pose.pt)
- **Run the code**:
```bash
python pose-estimate.py
# Options:
python pose-estimate.py --source "your-video.mp4" --device cpu # For CPU
python pose-estimate.py --source 0 --view-img # For Webcam
python pose-estimate.py --source "rtsp://your-ip" --device 0 --view-img # For LiveStream
```
- Output: The processed video will be saved as **your-file-keypoint.mp4**
### RESULTS
Football Match
Cricket Match
FPS & Time Comparison
Live Stream
### References
- YOLOv7 Repo: https://github.com/WongKinYiu/yolov7
- Ultralytics: https://github.com/ultralytics/yolov5
### 📖 Articles
- [YOLOv7 Training Guide](https://medium.com/augmented-startups/yolov7-training-on-custom-data-b86d23e6623)
- [Computer Vision Roadmap](https://medium.com/augmented-startups/roadmap-for-computer-vision-engineer-45167b94518c)