wip

1 files changed, 14 insertions, 7 deletions

diff --git a/SD-VBS/benchmarks/stitch/src/c/script_stitch.c b/SD-VBS/benchmarks/stitch/src/c/script_stitch.c
index 472e231..ee68e70 100644
--- a/SD-VBS/benchmarks/stitch/src/c/script_stitch.c
+++ b/SD-VBS/benchmarks/stitch/src/c/script_stitch.c
@@ -5,7 +5,8 @@ Author: Sravanthi Kota Venkata
 #include "stitch.h"
 #include <malloc.h>
 #include "extra.h"
-#define STITCH_MEM 1<<30
+// Valgrind says we use 250Mb of heap - round up to 512Mb
+#define STITCH_MEM 1<<29
 int main(int argc, char* argv[])
 {
     SET_UP
@@ -16,6 +17,7 @@ int main(int argc, char* argv[])
     I2D *Icur;
     int i, j;
     char im1[100], im2[100];
+    x = y = v = Fcur = int1 = int2 = interestPnts = NULL;
 
     scanf("%s", im1);
     Icur = readImage(im1);
@@ -23,6 +25,11 @@ int main(int argc, char* argv[])
     cols = Icur->width;
 
     for_each_job {
+        fFreeHandle(v);
+        fFreeHandle(interestPnts);
+        fFreeHandle(int1);
+        fFreeHandle(int2);
+        fFreeHandle(Fcur);
         v = harris(Icur);
         interestPnts = getANMS(v, 24);
         int1 = fMallocHandle(interestPnts->height, 1);
@@ -34,12 +41,6 @@ int main(int argc, char* argv[])
         }
 
         Fcur = extractFeatures(Icur, int1, int2);  
-        iFreeHandle(Icur);
-        fFreeHandle(v);
-        fFreeHandle(interestPnts);
-        fFreeHandle(int1);
-        fFreeHandle(int2);
-        fFreeHandle(Fcur);
     }
 
 #ifdef CHECK
@@ -56,5 +57,11 @@ int main(int argc, char* argv[])
     }
 #endif
     WRITE_TO_FILE
+    fFreeHandle(Icur);
+    fFreeHandle(v);
+    fFreeHandle(interestPnts);
+    fFreeHandle(int1);
+    fFreeHandle(int2);
+    fFreeHandle(Fcur);
     return 0;
 }