caffe的大多数层是由c++写成的,借助于c++的高效性,网络可以快速训练。但是我们有时候需要自己写点输入层以应对各种不同的数据输入,比如你因为是需要在图像中取块而不想写成LMDB,这时候可以考虑使用python直接写一个层。而且输入层不需要GPU加速,所需写起来也比较容易。
方法步骤比较固定,很多优秀的博主已经写的很clear了:
reference:
https://blog.youkuaiyun.com/haima1998/article/details/79066084
https://www.jianshu.com/p/e05d1b210fcb
https://blog.youkuaiyun.com/thesby/article/details/51264439
https://blog.youkuaiyun.com/Yan_Joy/article/details/55513627
本文介绍如何在Caffe中使用Python编写自定义输入层,以适应不同数据输入需求,如图像取块等场景。文章提供了实现步骤,并推荐了几篇优秀的参考博客,帮助读者理解和实践。
638

被折叠的 条评论
为什么被折叠?



