| | |
| | | DROPOUT, |
| | | CROP, |
| | | ROUTE, |
| | | COST |
| | | COST, |
| | | NORMALIZATION, |
| | | REGION, |
| | | AVGPOOL |
| | | } LAYER_TYPE; |
| | | |
| | | typedef enum{ |
| | |
| | | int batch; |
| | | int inputs; |
| | | int outputs; |
| | | int truths; |
| | | int h,w,c; |
| | | int out_h, out_w, out_c; |
| | | int n; |
| | | int groups; |
| | | int size; |
| | | int side; |
| | | int stride; |
| | | int pad; |
| | | int crop_width; |
| | | int crop_height; |
| | | int sqrt; |
| | | int flip; |
| | | float angle; |
| | | float saturation; |
| | | float exposure; |
| | | int softmax; |
| | | int classes; |
| | | int coords; |
| | | int background; |
| | |
| | | int objectness; |
| | | int does_cost; |
| | | int joint; |
| | | int noadjust; |
| | | |
| | | float alpha; |
| | | float beta; |
| | | float kappa; |
| | | |
| | | float coord_scale; |
| | | float object_scale; |
| | | float noobject_scale; |
| | | float class_scale; |
| | | |
| | | int dontload; |
| | | |
| | | float probability; |
| | | float scale; |
| | | |
| | | int *indexes; |
| | | float *rand; |
| | | float *cost; |
| | |
| | | int * input_sizes; |
| | | float * delta; |
| | | float * output; |
| | | float * squared; |
| | | float * norms; |
| | | |
| | | #ifdef GPU |
| | | int *indexes_gpu; |
| | |
| | | float * output_gpu; |
| | | float * delta_gpu; |
| | | float * rand_gpu; |
| | | float * squared_gpu; |
| | | float * norms_gpu; |
| | | #endif |
| | | } layer; |
| | | |
| | | void free_layer(layer); |
| | | |
| | | #endif |