[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文件夹内。
测试
测试结果部分截图: