[CVPR 2019]类别级物体6自由度位姿估计代码训练、检测与评估

Normalized Object Coordinate Space for Category-Level 6D Object Pose and Size Estimation 训练、检测与评估

本文作者：耗子
github：https://github.com/lh641446825
知乎：https://www.zhihu.com/people/hao-zi-meng-jian-mao/activities
博客园：https://www.cnblogs.com/lh641446825/
欢迎关注！

环境要求：

• cuda80（作者没有提到这个问题，但是我在cuda90上运行会报错）
• Python 3.5
• Tensorflow 1.3.0
• Keras 2.1.5
• cPickle

训练：

       训练时会报错：ImportError：No module named ‘visualize’。

       原因是作者没有将Mask-RCNN代码中的visualize.py文件打包，但是从pycharm上可以看到import visualize是灰色的，作者并没有用到，所以我就先把import visualize注释掉了。
预训练的COCO权重mask_rcnn_coco.h5作者也没有提供，下载地址：mask_rcnn_coco.h5下载之后放在logs文件夹内。

测试

       测试结果部分截图：

欢迎关注和转载！