Skip to content

Commit c8976c2

Browse files
committed
Import batch_detections, and network_predict_batch
1 parent 73822d7 commit c8976c2

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/libdarknet/__init__.pxd

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,13 @@ cdef extern from "darknet.h":
5151

5252
void free_detections(detection* detections, int len)
5353

54+
ctypedef struct det_num_pair:
55+
int num;
56+
detection* dets;
57+
58+
void free_batch_detections(det_num_pair* det_num_pairs, int len)
59+
60+
5461
void do_nms_sort(detection* detections, int len, int num_classes, float thresh)
5562
void do_nms_obj(detection* detections, int len, int num_classes, float thresh)
5663

@@ -68,6 +75,8 @@ cdef extern from "darknet.h":
6875
int network_output_size(network* self);
6976
float* network_predict(network, float* input)
7077
float* network_predict_image(network*, image)
78+
7179
detection* get_network_boxes(network* self, int width, int height, float thresh, float hier_thresh, int* map, int relative, int* out_len, int letter)
80+
det_num_pair* network_predict_batch(network* self, image, int batch_size, int width, int height, float thresh, float hier_thresh, int* map, int relative, int letter)
7281

7382

0 commit comments

Comments
 (0)