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