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

测试
       测试结果部分截图:

 

欢迎关注和转载!

posted @ 2019-09-03 16:54  耗子梦见猫  阅读(823)  评论(0编辑  收藏  举报