我想使用 Caffe 提取图像特征。但是,处理图像的时间太长,因此我正在寻找优化速度的方法。
我注意到的一件事是,我正在使用的网络定义在我从中读取结果的那一层之上有四个额外的层(并且没有反馈信号,因此删除它们应该是安全的)。
我试图从定义文件中删除它们,但完全没有效果。我想我可能还需要删除文件中包含预训练权重的相应部分。然而,这是一个二进制文件(一个原型(prototype)缓冲区),因此编辑它并不那么容易。
您认为移除四层可能会对净性能产生深远影响吗?
如果是这样,那么我如何熟悉文件内容以便我可以对其进行编辑以及我如何知道要删除哪些部分?
最佳答案
首先,我认为移除二进制权重不会有任何影响。
其次,您可以使用 python 界面轻松完成:参见 this tutorial .
最后但同样重要的是,您是否尝试过运行 caffe time
衡量你的网络的表现?这可能会帮助您确定计算的瓶颈。
附言, 你可能会发现 this thread也相关。
关于c++ - 减少 Caffe 网络模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32505458/