caffe
文章平均质量分 88
enmengyi
这个作者很懒,什么都没留下…
展开
-
Caffe中卷积的实现
方法:将卷积转化为矩阵相乘 两步: (1)调用im2col函数,将图像(feature map)转化为矩阵 (2)调用GEMM函数,做实际的运算优点:逻辑简单,易于实现缺点:耗费大量内存feature map每一个卷积位置(卷积核将要滑过的位置),展开成一个C∗K∗KC*K*K维的行向量继续展开……按照这个方式展开完毕之后,得到一个矩阵:H∗WH*W行(因为有H∗WH*W个卷积位置),C∗原创 2017-05-11 01:45:28 · 557 阅读 · 0 评论 -
Caffe中使用HDF5制作多标签数据
在Caffe中,如果使用LMDB数据格式的话,默认是只支持“图像+整数单标签”这种形式的数据的。如果训练网络需要一些其他形式的数据或标签(如浮点数据,多标签等等),可以将其制作成HDF5格式。PS:HDF5数据格式比较灵活,但缺点是占用空间较大。笔者曾经试着把229多张512x512的图像制作成一个HDF5文件,可以有1.4G. 因此建议图像文件的话,最好还是用LMDB格式,快速且节省空间;在网络定原创 2017-09-06 16:40:39 · 2555 阅读 · 3 评论