【深度学习】在caffe中添加新层(layers)

在上一篇文章中【深度学习】caffe中那些layers,我们详细地介绍了深度学习框架caffe中的基本构成单元——layer。深度学习发展至今,已有最开始的conv层、pooling层等少数的几种layer,发展到了拥有数十种layer。caffe提供了这其中大部分层的CPU和GPU(CUDA)实现(如下图所示)。然而,在我们科研过程中,当灵感闪现,自己想到了更好的idea,要去实现新层要怎么办呢?本文就介绍了如何在caffe中实现自己的新层(layers)。

【深度学习】在caffe中添加新层(layers)_第1张图片

你可能感兴趣的:(深度学习,caffe)