Minor fix

This commit is contained in:
AlexeyAB
2020-04-22 20:49:27 +03:00
parent 88f28f7fcc
commit b573eab63f
6 changed files with 92 additions and 12 deletions

View File

@ -831,10 +831,10 @@ typedef struct detection{
} detection;
// network.c -batch inference
typedef struct detNumPair {
typedef struct det_num_pair {
int num;
detection *dets;
} detNumPair, *pdetNumPair;
} det_num_pair, *pdet_num_pair;
// matrix.h
typedef struct matrix {
@ -945,7 +945,9 @@ LIB_API void diounms_sort(detection *dets, int total, int classes, float thresh,
LIB_API float *network_predict(network net, float *input);
LIB_API float *network_predict_ptr(network *net, float *input);
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);
LIB_API void free_batch_detections(det_num_pair *det_num_pairs, int n);
LIB_API void fuse_conv_batchnorm(network net);
LIB_API void calculate_binary_weights(network net);
LIB_API char *detection_to_json(detection *dets, int nboxes, int classes, char **names, long long int frame_id, char *filename);