trrahul / densepose-video

Code to run densepose on video with detectron. https://github.com/facebookresearch/Detectron
GNU General Public License v3.0
62 stars 18 forks source link

Which video format do it take and do it give output in the form of json? #15

Open arshikasoni opened 2 years ago

arshikasoni commented 2 years ago

How can we take output in U-V format? and why is it giving output of just last frame

image

trrahul commented 2 years ago

Could you post the output log? You can change the ffmpeg params to get the output you want. https://github.com/trrahul/densepose-video/blob/2157512c7604da1f00ef9a2065d4c81deff382df/infer_vid.py#L145

arshikasoni commented 2 years ago

I am running in colab so here is the output

image

also I am getting just the last frame of the video like as shown in the above video, I want to know how can we get some u-v cordinates of some body parts from where we can calculate angles of some body parts.

trrahul commented 2 years ago

You can see it on the terminal.

arshikasoni commented 2 years ago

Can't we do something like we get some kind of json or csv of the body coordinates, so that we can calculate angles between body parts?

trrahul commented 2 years ago

You can see the output logs in stdout, normally in terminal output if you run the script from a terminal. This script writes the visualizations on the input frame and saves it as an image, after it creates a video with the frame images. You can check this code to make use of body_uv.

https://github.com/trrahul/densepose-video/blob/2157512c7604da1f00ef9a2065d4c81deff382df/vis.py#L384

trrahul commented 2 years ago

Also please note that this code was last updated 4 years ago

arshikasoni commented 2 years ago

This is running on detectron or detection2?