淘先锋技术网

首页 1 2 3 4 5 6 7

caffe的大多数层是由c++写成的,借助于c++的高效性,网络可以快速训练。但是我们有时候需要自己写点输入层以应对各种不同的数据输入,比如你因为是需要在图像中取块而不想写成LMDB,这时候可以考虑使用python直接写一个层。而且输入层不需要GPU加速,所需写起来也比较容易。

方法步骤比较固定,很多优秀的博主已经写的很clear了:

reference:

https://blog.csdn.net/haima1998/article/details/79066084

https://www.jianshu.com/p/e05d1b210fcb

https://blog.csdn.net/thesby/article/details/51264439

https://blog.csdn.net/Yan_Joy/article/details/55513627