darknet源碼解析:networ結構體之input_pinned_gpu

更多darknet源代碼學習筆記,參看:darknet源碼學習:預測分類函數float *network_predict_gpu(network net, float *input)

parse_network_cfg_custom函數在解析cfg文件,parse_network_cfg()中調用

network parse_network_cfg(char *filename)
{
    return parse_network_cfg_custom(filename, 0, 0);
}

input_pinned_gpu在parse_network_cfg_custom()中初始化

 int size = get_network_input_size(net) * net.batch;
 net.input_pinned_cpu = (float*)calloc(size, sizeof(float));

 

發佈了373 篇原創文章 · 獲贊 151 · 訪問量 33萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章