0x00
关于使用C++接口来提取特征,caffe官方提供了一个extract_features.cpp的例程,但是这个文件的输入是blob数据,即使输入层使用的是ImageData,也需要在deploy.prototxt中指定图片的位置,很不方便。
如果想要使用opencv来读取一个图片,然后用caffe训练好的model提取特征,就需要对输入层进行改写。另外官方例程默认的输出是leveldb格式,我们也可以获取float类型的多维特征(数组),这样集成到我们的项目中更灵活。
0x01
首先我们需要改写deploy.prototxt的输入层为"MemoryData":
layer { name: "data" type: "MemoryData" top: "data" top: "label" memory_data_param{ batch_size:1 channels:3 height:100 width:100 } }
在之前的训练中可能使用的是"ImageData"、"Data"之类的,现在改成MemoryData不影响。
0x02
我准备提取的层的名字是"res5_6",就是"InnerProduct"的