Mixup bug fix

6 years ago · 3ff27b1cdd
parent 57baf1a487
commit 3ff27b1cdd
1 changed files with 2 additions and 0 deletions
--- a/src/data.c
+++ b/src/data.c
@ -897,6 +897,7 @@ data load_data_detection(int n, char **paths, int m, int w, int h, int c, int bo
                //wait_until_press_key_cv();
                blend_images_cv(ai, 0.5, old_img, 0.5);
                blend_truth(truth, boxes, d.y.vals[i]);
+                free_image(old_img);
            }

            d.X.vals[i] = ai.data;
@ -1036,6 +1037,7 @@ data load_data_detection(int n, char **paths, int m, int w, int h, int c, int bo
                //wait_until_press_key_cv();
                blend_images(sized, 0.5, old_img, 0.5);
                blend_truth(truth, boxes, d.y.vals[i]);
+                free_image(old_img);
            }

            d.X.vals[i] = sized.data;