From c53e03348c65462bcba33f6352087dd6b9268e8f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 16 Sep 2015 21:12:10 +0000
Subject: [PATCH] yolo working w/ regions
---
src/compare.c | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/compare.c b/src/compare.c
index 9b6d6bf..0408f80 100644
--- a/src/compare.c
+++ b/src/compare.c
@@ -175,8 +175,8 @@
image im1 = load_image_color(box1.filename, net.w, net.h);
image im2 = load_image_color(box2.filename, net.w, net.h);
float *X = calloc(net.w*net.h*net.c, sizeof(float));
- memcpy(X, im1.data, im1.w*im1.h*im1.c);
- memcpy(X+im1.w*im1.h*im1.c, im2.data, im2.w*im2.h*im2.c);
+ memcpy(X, im1.data, im1.w*im1.h*im1.c*sizeof(float));
+ memcpy(X+im1.w*im1.h*im1.c, im2.data, im2.w*im2.h*im2.c*sizeof(float));
float *predictions = network_predict(net, X);
free_image(im1);
--
Gitblit v1.10.0