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萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章