天天看點

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));
           

繼續閱讀