Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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: 3 months 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: 2024-11-04T09:36:27.000Z (4 months ago)
- Last Synced: 2024-11-04T10:30:08.875Z (4 months ago)
- Topics: medium-article, opencv, opencv-python, pose-estimation, yolov7
- Language: Python
- Homepage:
- Size: 3.64 MB
- Stars: 342
- Watchers: 6
- Forks: 82
- Open Issues: 9
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE
Awesome Lists containing this project
- awesome-yolo-object-detection - RizwanMunawar/yolov7-pose-estimation - pose-estimation?style=social"/> : YOLOv7 Pose estimation using OpenCV, PyTorch. (Applications)
- awesome-yolo-object-detection - RizwanMunawar/yolov7-pose-estimation - pose-estimation?style=social"/> : YOLOv7 Pose estimation using OpenCV, PyTorch. (Applications)
README
# yolov7-pose-estimation
### New Features
- Added Support for Comparision of (FPS & Time) Graph
- How to run Code in Google Colab
- Code can run on Both (CPU & GPU)
- Video/WebCam/External Camera/IP Stream Supported
### Coming Soon
- Development of streamlit dashboard for Pose-Estimation
### Steps to run Code
- If you are using google colab then you will first need to mount the drive with mentioned command first, (Windows or Linux users) both can skip this step.
```
from google.colab import drive
drive.mount("/content/drive")
```
- Clone the repository.
```
git clone https://github.com/RizwanMunawar/yolov7-pose-estimation.git
```
- Goto the cloned folder.
```
cd yolov7-pose-estimation
```
- Create a virtual envirnoment (Recommended, If you dont want to disturb python packages)
```
### For Linux Users
python3 -m venv psestenv
source psestenv/bin/activate
### For Window Users
python3 -m venv psestenv
cd psestenv
cd Scripts
activate
cd ..
cd ..
```
- Upgrade pip with mentioned command below.
```
pip install --upgrade pip
```
- Install requirements with mentioned command below.
```
pip install -r requirements.txt
```
- Download yolov7 pose estimation weights from [link](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6-pose.pt) and move them to the working directory {yolov7-pose-estimation}
- Run the code with mentioned command below.
```
python pose-estimate.py
#if you want to change source file
python pose-estimate.py --source "your custom video.mp4"
#For CPU
python pose-estimate.py --source "your custom video.mp4" --device cpu
#For GPU
python pose-estimate.py --source "your custom video.mp4" --device 0
#For View-Image
python pose-estimate.py --source "your custom video.mp4" --device 0 --view-img
#For LiveStream (Ip Stream URL Format i.e "rtsp://username:pass@ipaddress:portno/video/video.amp")
python pose-estimate.py --source "your IP Camera Stream URL" --device 0 --view-img
#For WebCam
python pose-estimate.py --source 0 --view-img
#For External Camera
python pose-estimate.py --source 1 --view-img
```
- Output file will be created in the working directory with name ["your-file-name-without-extension"+"_keypoint.mp4"]
#### RESULTS
Football Match Pose-Estimation
Cricket Match Pose-Estimation
FPS and Time Comparision
Live Stream Pose-Estimation
#### References
- https://github.com/WongKinYiu/yolov7
- https://github.com/augmentedstartups/yolov7
- https://github.com/augmentedstartups
- https://learnopencv.com/yolov7-object-detection-paper-explanation-and-inference/
- https://github.com/ultralytics/yolov5
#### My Medium Articles
- https://medium.com/augmented-startups/yolov7-training-on-custom-data-b86d23e6623
- https://medium.com/augmented-startups/roadmap-for-computer-vision-engineer-45167b94518c
- https://medium.com/augmented-startups/yolor-or-yolov5-which-one-is-better-2f844d35e1a1
- https://medium.com/augmented-startups/train-yolor-on-custom-data-f129391bd3d6
- https://medium.com/augmented-startups/develop-an-analytics-dashboard-using-streamlit-e6282fa5e0f