darknet源码学习：get_network_input_size

更多darknet源代码学习笔记，参看：darknet源码学习：预测分类函数float *network_predict_gpu(network net, float *input)

get_network_input_size获得网络输入层大小，源代码见  

if (net.gpu_index != cuda_get_device())
        cuda_set_device(net.gpu_index);

从network定义network结构体可以知道，

结构体中有一个变量 

layer* layers;

其中layer定义见：layer结构体

inputs是当前层输入维度=w*h*c

w:图像宽度

h:图像高度

c:图像通道

network定义见：结构体network定义。

 

cudaError_t cudaSetDevice(int dev)

将dev记录为活动主线程将执行设备码的设备。
cuda_get_device，获得活动主线程将执行的设备码的设备