caffe做部署是YYDS!
caffe 源码阅读笔记
·30 字·1 分钟
99 - 这篇文章属于一个选集。
99 - 这篇文章属于一个选集。
相关文章
caffe 源码学习笔记(11) argmax layer
·1300 字·3 分钟
背景 # 似乎没什么背景,继续看 caffe 代码。
argmax 的作用是返回一个 blob 某个维度(或者 batch_size 之后的维度)的 top_k 的 index(或者 pair(index, value))。
caffe 源码学习笔记(10) eltwise layer
·988 字·2 分钟
背景 # 这个 layer 和 reduce layer 有一些相似,就干脆一起看了。 作用是输入至少两个 blob,然后对每个 blob 中的元素做一些运算,最后得到一个 blob。
caffe 源码学习笔记(9) reduce layer
·1251 字·3 分钟
背景 # 其实没什么背景,继续啃 caffe 代码而已 2333。
reduce layer 其实就是做 reduce 操作,把一个任意 shape 的 blob 通过某种运算变成一个 scalar。
caffe 源码学习笔记(8) loss function
·1095 字·3 分钟
背景 # 虽然不太care 训练的过程,但是由于容易看懂的layer都看得差不多了 所以打算看一下这些loss function.
caffe 源码学习笔记(7) slice layer
·1121 字·3 分钟
背景 # OCR 组那边有个 shufflenet 的网络,里面有个 pytorch op 叫 chunk,转成的 onnx 对应的 op 是 split。
作用是:
Split a tensor into a list of tensors, along the specified ‘axis’. Lengths of the parts can be specified using argument ‘split’. Otherwise, the tensor is split to equal sized parts.