search

firephinx / openpose_ros

ROS Catkin package wrapper for the OpenPose library from https://github.com/CMU-Perceptual-Computing-Lab/openpose.
55 stars 44 forks source link

is there a way to use the pytorch models instead of the caffe ones ?? #21

Closed ymohamed08 closed 4 years ago

ymohamed08 commented 4 years ago

i am trying to use the openPose model in https://github.com/Hzzone/pytorch-openpose, is there a way to do so easily ??

firephinx commented 4 years ago

It should be fairly straightforward. Just combine the demo.py code from the repository: https://github.com/Hzzone/pytorch-openpose/blob/master/demo.py with code to subscribe to ROS images like this one: http://wiki.ros.org/cv_bridge/Tutorials/ConvertingBetweenROSImagesAndOpenCVImagesPython

ymohamed08 commented 4 years ago

thank you