Caffe學習之自定義創建新的Layer層

轉載:https://blog.csdn.net/BVL10101111/article/details/74837156 一.創建.hpp文件 1.添加你的layer頭文件置於 include/caffe/layers/ 下,比如include/caffe/layers/your_layer.hpp 2.your_layer繼承選擇繼承layer.hpp, common_layers.hpp,
相關文章
相關標籤/搜索