Code is available in GitHub,
I taught these AI to dance on In-My-Feelings song by Drake using PoseNet and Tensorflow.js
I have used the Pose Detection model called PoseNet implemented in Tensorflow.js. The original repository can be found in https://github.com/tensorflow/tfjs-models/tree/master/posenet.
I have modified the codes to enable the following
- Loads a pre-trained PoseNet model
- Runs PoseNet model on any video saved on computer
- Detects 16 set of key-points from sixteen parts of the human body from the video
- Canvas is being recorded and saved as .mp4!
The input video, output video and background image can be found in ‘video’ directory.