[CVPR 2019]NOCS代码训练、检测与评估

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

本文作者：耗子

CSDN：https://blog.csdn.net/lh641446825?t=1

github：https://github.com/lh641446825

知乎：https://www.zhihu.com/people/hao-zi-meng-jian-mao/activities

欢迎关注！

环境要求：

• 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文件夹内。

测试
       测试结果部分截图：

欢迎关注和转载！