diff --git a/src/imgaug/cpu/host_color_model_conversions.hpp b/src/imgaug/cpu/host_color_model_conversions.hpp index 78c779685..d0b2cdfdf 100644 --- a/src/imgaug/cpu/host_color_model_conversions.hpp +++ b/src/imgaug/cpu/host_color_model_conversions.hpp @@ -204,6 +204,11 @@ RppStatus vignette_host(T* srcPtr, RppiSize srcSize, T* dstPtr, } compute_multiply_host(srcPtr, maskFinal, srcSize, dstPtr, channel); - + + free(maskFinal); + free(kernelRows); + free(kernelColumns); + free(mask); + return RPP_SUCCESS; } \ No newline at end of file diff --git a/src/imgaug/cpu/host_image_augmentations.hpp b/src/imgaug/cpu/host_image_augmentations.hpp index 27b12c998..0ad8721c6 100644 --- a/src/imgaug/cpu/host_image_augmentations.hpp +++ b/src/imgaug/cpu/host_image_augmentations.hpp @@ -29,7 +29,8 @@ RppStatus blur_host(T* srcPtr, RppiSize srcSize, T* dstPtr, rppiKernelSize.height = kernelSize; rppiKernelSize.width = kernelSize; convolve_image_host(srcPtrMod, srcSizeMod, dstPtr, srcSize, kernel, rppiKernelSize, chnFormat, channel); - + free(kernel); + free(srcPtrMod); return RPP_SUCCESS; } diff --git a/src/include/cpu/rpp_cpu_common.hpp b/src/include/cpu/rpp_cpu_common.hpp index 2c78904f9..5eb690d5d 100644 --- a/src/include/cpu/rpp_cpu_common.hpp +++ b/src/include/cpu/rpp_cpu_common.hpp @@ -406,23 +406,6 @@ inline RppStatus generate_sobel_kernel_host(Rpp32f* kernel, Rpp32u type) return RPP_SUCCESS; } - - - - - - - - - - - - - - - - - // Kernels for functions template @@ -589,6 +572,7 @@ inline RppStatus resize_crop_kernel_host(T* srcPtr, RppiSize srcSize, T* dstPtr, resize_kernel_host(srcPtrResize, srcSizeSubImage, dstPtr, dstSize, chnFormat, channel); + free(srcPtrResize); return RPP_SUCCESS; } @@ -727,7 +711,7 @@ inline RppStatus median_filter_kernel_host(T* srcPtrWindow, T* dstPtrPixel, Rppi std::sort(kernel, kernel + (kernelSize * kernelSize)); *dstPtrPixel = *(kernel + (((kernelSize * kernelSize) - 1) / 2)); - + free(kernel); return RPP_SUCCESS; } @@ -900,26 +884,6 @@ inline RppStatus non_max_suppression_kernel_host(T* srcPtrWindow, T* dstPtrPixel return RPP_SUCCESS; } - - - - - - - - - - - - - - - - - - - - // Convolution Functions template