From 527e2942e97714a5496bf3bc85ed0822bad04783 Mon Sep 17 00:00:00 2001 From: Dimitrii Date: Fri, 15 May 2020 16:05:02 +0300 Subject: [PATCH] clean up --- network.c | 4 +--- network.go | 1 - 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/network.c b/network.c index eaa6eb4..c137e90 100644 --- a/network.c +++ b/network.c @@ -14,10 +14,8 @@ struct network_box_result perform_network_detect(network *n, image *img, int cla sized = resize_image(*img, n->w, n->h); } struct network_box_result result = { NULL }; - float *X = sized.data; - network_predict(*n, X); + network_predict_ptr(n, sized.data); int nboxes = 0; - detection *dets = get_network_boxes(n, img->w, img->h, thresh, hier_thresh, 0, 1, &nboxes, letter_box); result.detections = get_network_boxes(n, img->w, img->h, thresh, hier_thresh, 0, 1, &result.detections_len, letter_box); if (nms) { do_nms_sort(result.detections, result.detections_len, classes, nms); diff --git a/network.go b/network.go index a4c71c7..31ef9fe 100644 --- a/network.go +++ b/network.go @@ -71,7 +71,6 @@ func (n *YOLONetwork) Detect(img *DarknetImage) (*DetectionResult, error) { startTime := time.Now() 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)) endTime := time.Now() - // defer C.free_detections(result.detections, result.detections_len) ds := makeDetections(img, result.detections, int(result.detections_len), n.Threshold, n.Classes, n.ClassNames) C.free_detections(result.detections, result.detections_len) endTimeOverall := time.Now()