RizwanMunawar / yolov7-pose-estimation

YOLOv7 Pose estimation using OpenCV, PyTorch
GNU General Public License v3.0
344 stars 82 forks source link

[Modifications] Plotting x, y displacement and replacing bounding box with a circle ? #11

Closed timtensor closed 1 year ago

timtensor commented 1 year ago

Hi, first of all great work done for combining pose estimation and object tracking. I am trying to modify the implementation to basically plot

  1. Heat map of few of the tracked objects to have some kind of the birds eye view . The idea was to plot if per frame and to have it visualized on top of the video stream (like realtime analysis) . Something like attached in the image itself .
  2. Is it possible to draw instead of a rectangular bounding box , some kind of a circular disc on the foot of the players for better and easier visualization?

Could you please provide some guidelines on it ? image

RizwanMunawar commented 1 year ago

@timtensor! My pleasure 1- I am not sure, but as far as my understanding you can create a separate OpenCV frame with a green background and plot of all the center points of objects on that. For more details, you can check the link mentioned below. https://www.linkedin.com/posts/muhammadrizwanmunawar_computervision-objectdetection-deeplearning-activity-6992501395882971136-OQ1D?utm_source=share&utm_medium=member_desktop

2- Yes, you will only need to find the exact position of the rectangle bounding box's bottom line, which you can take as the starting point of the circular disc.

timtensor commented 1 year ago

Thanks for the info ,

  1. I will check them out . I think the idea is just to plot velocity as you have shown . Is there any explanation of it or where the modifications have been made. I could not find much information.

2- Can you tell me where to modify the code in order to have the cirular disc visualization ?

RizwanMunawar commented 1 year ago

@timtensor! You will need to do all your work in plot.py or some in detect.py. For circular disc plotting, you will need to write the custom functions. You can take reference from the mentioned link below. https://github.com/roboflow-ai/notebooks/blob/main/notebooks/how-to-track-football-players.ipynb

RizwanMunawar commented 1 year ago

@timtensor I am closing this issue, as this is a question that has been answered last month. Feel free to open new issues, if you will face any problems with the code.