在上一篇博文中已经介绍了很多如何在caffe中使用python 层。有一些人问到如何使用这些层。
首先需要在编译caffe时,允许python layer,在配置文件中设置。
写好自己的层之后,需要
export $PYTHONPATH=./path/to/my_layers_dir:$PYTHONPATH
这样才能找到你的python文件。
然后module这个参数就是你的文件名,不需要后缀.py
param_str就是你需要的所有参数在这里设置,自己定义就好,这个其实就是一个字符串,目的是用来在层当中使用eval生成一个字典,然后参数就传进去了。
caffe中使用python layer
最新推荐文章于 2024-04-10 20:18:04 发布