diff --git a/include/darknet.h b/include/darknet.h index d72027cc45f..2517d3fbd05 100644 --- a/include/darknet.h +++ b/include/darknet.h @@ -1032,6 +1032,7 @@ LIB_API void diounms_sort(detection *dets, int total, int classes, float thresh, // network.h LIB_API float *network_predict(network net, float *input); LIB_API float *network_predict_ptr(network *net, float *input); +LIB_API void set_batch_network(network *net, int b); LIB_API detection *get_network_boxes(network *net, int w, int h, float thresh, float hier, int *map, int relative, int *num, int letter); LIB_API det_num_pair* network_predict_batch(network *net, image im, int batch_size, int w, int h, float thresh, float hier, int *map, int relative, int letter); LIB_API void free_detections(detection *dets, int n); diff --git a/src/network.h b/src/network.h index 7661c8ef806..9cfce4e940a 100644 --- a/src/network.h +++ b/src/network.h @@ -144,7 +144,7 @@ int get_predicted_class_network(network net); void print_network(network net); void visualize_network(network net); int resize_network(network *net, int w, int h); -void set_batch_network(network *net, int b); +//LIB_API void set_batch_network(network *net, int b); int get_network_input_size(network net); float get_network_cost(network net); //LIB_API layer* get_network_layer(network* net, int i);