summaryrefslogtreecommitdiff
path: root/utils/tuning/libtuning/smoothing.py
blob: de4d920cfe9cef81f05aaf6a42dbaebc7e859353 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# SPDX-License-Identifier: GPL-2.0-or-later
#
# Copyright (C) 2022, Paul Elder <paul.elder@ideasonboard.com>
#
# 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')