| | |
| | | #ifndef NETWORK_H |
| | | #define NETWORK_H |
| | | |
| | | #include "image.h" |
| | | #include "layer.h" |
| | | |
| | | #ifdef __cplusplus |
| | | extern "C" { |
| | | #endif |
| | | |
| | | #include "image.h" |
| | | #include "data.h" |
| | | #include "tree.h" |
| | | |
| | | typedef enum { |
| | | CONSTANT, STEP, EXP, POLY, STEPS, SIG, RANDOM |
| | |
| | | int num_steps; |
| | | int burn_in; |
| | | |
| | | int adam; |
| | | float B1; |
| | | float B2; |
| | | float eps; |
| | | |
| | | int inputs; |
| | | int h, w, c; |
| | | int max_crop; |
| | |
| | | float hue; |
| | | |
| | | int gpu_index; |
| | | tree *hierarchy; |
| | | |
| | | #ifdef GPU |
| | | float **input_gpu; |
| | |
| | | int get_network_nuisance(network net); |
| | | int get_network_background(network net); |
| | | |
| | | #ifdef __cplusplus |
| | | } |
| | | #endif |
| | | |
| | | #endif |
| | | |