darknet源码解析:layer结构体之forward_gpu

darknet源码学习:forward_network_gpu

layer结构体中forward_gpu定义:

void(*forward_gpu)   (struct layer, struct network_state);

不同网络层forward_gpu指向不同函数:

darknet源码解析:network结构体之n中,可以看出cfg文件中包含四种类型网络层:CONVOLUTIONAL,SOFTMAX,AVGPOOL,MAXPOOL。

darknet源码解析:forward_convolutional_layer_gpu

darknet源码解析:forward_maxpool_layer_gpu

darknet源码解析:forward_softmax_layer_gpu

darknet源码解析:forward_avgpool_layer_gpu

 

发布了373 篇原创文章 · 获赞 151 · 访问量 33万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章