在py-faster-rcnn中用python自定义caffe层

在demo.py中用sys.path输出看到,PYTHONPATH中有lib文件夹。

所以说在lib里直接定义的new_layer.py文件是可以被找到的。

如果在lib中新建文件夹A,再在A中定义new_layer.py文件就不能被直接找到了,如果要找到,需要去修改下脚本文件。

比如说我现在要执行./experiments/scripts/faster_rcnn_end2end.sh 0 VGG16 pascal_voc的话,就去faster_rcnn_end2end.sh文件中,加上:

export PYTHONPATH=$PYTHONPATH:lib/A

也就是把lib/A添加到PYTHONPATH中了,现在再在models/pascal_voc/VGG16/faster_rcnn_end2end/train.prototxt里面加上这个用python写的caffe层就没有问题了,不会找不到了:

layer {

    name: '***'

    type: 'Python'

    bottom: '**'

    top: '**'

    python_param {

        module: 'new_layer'              #不需要.py后缀哦

        layer: 'NewLayer'                 #文件中的类名

    }

}


问了py-F-RCN的作者,其实最简单的就是在自己创建的文件夹下,新建一个空白的"__init__.py"文件,然后调用时用:module: 'A.new_layer' 就行了。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 1,下载编译fast-rcnn fast-rcnn下载地址 Clone the Fast R-CNN reposi...
    Mahxn0阅读 1,202评论 2 1
  • 作为一名HR,我经常要为公司招聘实习生。刚开始工作的时候,我总会体谅实习生工资低,经常主动提醒他们:中午可以不用和...
    澄紫橙阅读 526评论 6 6
  • 坐上歲月回程的晚車 冷色的往事 隨著車輪輾過的夜雨 擱淺在夢岸 消失了 一片星空 因為 不用再想你 天亮了 漫天朝...
    秋鳶子阅读 478评论 0 1
  • 还是不能让自己过得太舒服,还是想给找点事情做。即使做之前我不那么愿意,可是后来慢慢明白,比起没做好,不尝试更让人难...
    野麦_阅读 223评论 0 1
  • 耳畔缭绕曲音,我独白着:你的孤独虽败犹荣。我敲击着键盘在搜索引擎上浏览着,他的文字讲述着我的心情。我突然很想了解当...
    遮個刹手不太冷阅读 192评论 0 0