OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

feat: Move set_batch_network to public API (#8782)

This commit is contained in:
Tobit Flatscher 2023-08-02 07:54:08 +01:00 committed by GitHub
parent 5915c2e09a
commit d2c3788539
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/darknet.h
View File

@ -1032,6 +1032,7 @@ LIB_API void diounms_sort(detection *dets, int total, int classes, float thresh,
// network.h // network.h
LIB_API float *network_predict(network net, float *input); LIB_API float *network_predict(network net, float *input);
LIB_API float *network_predict_ptr(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 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 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); LIB_API void free_detections(detection *dets, int n);

2
src/network.h
View File

@ -144,7 +144,7 @@ int get_predicted_class_network(network net);
void print_network(network net); void print_network(network net);
void visualize_network(network net); void visualize_network(network net);
int resize_network(network *net, int w, int h); 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); int get_network_input_size(network net);
float get_network_cost(network net); float get_network_cost(network net);
//LIB_API layer* get_network_layer(network* net, int i); //LIB_API layer* get_network_layer(network* net, int i);