src/network.h @@ -34,11 +34,17 @@ float *scales; int *steps; int num_steps; int burn_in; int inputs; int h, w, c; int max_crop; int min_crop; float angle; float exposure; float saturation; int gpu_index; #ifdef GPU float **input_gpu;
