ROS:img2pose
通过6DOF人脸姿势估计进行人脸对齐和检测
该存储库为img2pose神经网络提供了一个ROS包装节点,用于在多个面部上进行六个自由度(6DoF)检测,而无需事先进行面部检测。 此存储库中包含的模型已在WIDER FACE数据集中进行了训练。 原始模型和培训说明可在项目。
在Intel 和GeForce GTX1060 / 6Gb上以运行
安装
将此存储库克隆到catkin工作区中,并在运行catkin_make之前使用catkin_make安装依赖catkin_make 。
pip3 install -r requirements.txt
用法
当前有一个包装器节点,用于加载模型,订阅sensor_msgs / Image主题,在图像回调函数中运行预测并以彩色轴的形式将6DOF头部姿势渲染到框架上。
rosrun ros_img2pose img2po
2022-05-16 16:07:17
151.08MB
Python
1