我们已基于此工作发布了新项目的代码和arXiv预印本,该代码和预印本是基于Wang等人撰写的论文“DenseFusion:通过迭代密集融合进行6D对象姿势估计”的实现。该模型接受RGB-D图像作为输入,预测每个对象在帧中的6D姿势。代码在Python中实现,包括DenseFusion模型、迭代优化模型和SegNet语义分割模型的完整实现。ROS代码未包含在内。支持Python 2.7/3.5/3.6,并建议使用PyTorch 0.4.1及以上版本。