Skip to content

Commit 420a747

Browse files
authored
Merge pull request #7 from LdDl/mleak
Closes #6
2 parents 8105b8f + 527e294 commit 420a747

File tree

2 files changed

+1
-4
lines changed

2 files changed

+1
-4
lines changed

network.c

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,8 @@ struct network_box_result perform_network_detect(network *n, image *img, int cla
1414
sized = resize_image(*img, n->w, n->h);
1515
}
1616
struct network_box_result result = { NULL };
17-
float *X = sized.data;
18-
network_predict(*n, X);
17+
network_predict_ptr(n, sized.data);
1918
int nboxes = 0;
20-
detection *dets = get_network_boxes(n, img->w, img->h, thresh, hier_thresh, 0, 1, &nboxes, letter_box);
2119
result.detections = get_network_boxes(n, img->w, img->h, thresh, hier_thresh, 0, 1, &result.detections_len, letter_box);
2220
if (nms) {
2321
do_nms_sort(result.detections, result.detections_len, classes, nms);

network.go

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,6 @@ func (n *YOLONetwork) Detect(img *DarknetImage) (*DetectionResult, error) {
7171
startTime := time.Now()
7272
result := C.perform_network_detect(n.cNet, &img.image, C.int(n.Classes), C.float(n.Threshold), C.float(n.hierarchalThreshold), C.float(n.nms), C.int(0))
7373
endTime := time.Now()
74-
// defer C.free_detections(result.detections, result.detections_len)
7574
ds := makeDetections(img, result.detections, int(result.detections_len), n.Threshold, n.Classes, n.ClassNames)
7675
C.free_detections(result.detections, result.detections_len)
7776
endTimeOverall := time.Now()

0 commit comments

Comments
 (0)