From d2c3788539bb73d3cfcdb7a1d5fa6b29ae1ec0e1 Mon Sep 17 00:00:00 2001 From: Tobit Flatscher <53856473+2b-t@users.noreply.github.com> Date: Wed, 2 Aug 2023 07:54:08 +0100 Subject: [PATCH] feat: Move set_batch_network to public API (#8782) --- include/darknet.h | 1 + src/network.h | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/include/darknet.h b/include/darknet.h index d72027cc..2517d3fb 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 7661c8ef..9cfce4e9 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);