db99d96663
Implement math helpers for libtuning. This includes: - Average, a wrapper class for numpy averaging functions - Gradient, a class that represents gradients, for distributing and mapping - Smoothing, a wrapper class for cv2 smoothing functions Signed-off-by: Paul Elder <paul.elder@ideasonboard.com> Reviewed-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
24 lines
606 B
Python
24 lines
606 B
Python
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
#
|
|
# Copyright (C) 2022, Paul Elder <paul.elder@ideasonboard.com>
|
|
#
|
|
# smoothing.py - Wrapper for cv2 smoothing functions to enable duck-typing
|
|
|
|
import cv2
|
|
|
|
|
|
# @brief Wrapper for cv2 smoothing functions so that they can be duck-typed
|
|
class Smoothing(object):
|
|
def __init__(self):
|
|
pass
|
|
|
|
def smoothing(self, src):
|
|
raise NotImplementedError
|
|
|
|
|
|
class MedianBlur(Smoothing):
|
|
def __init__(self, ksize):
|
|
self.ksize = ksize
|
|
|
|
def smoothing(self, src):
|
|
return cv2.medianBlur(src.astype('float32'), self.ksize).astype('float64')
|