File: smoothing.py

package info (click to toggle)
libcamera 0.5.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 14,380 kB
  • sloc: cpp: 81,540; python: 18,107; ansic: 12,608; sh: 1,258; makefile: 56
file content (24 lines) | stat: -rw-r--r-- 591 bytes parent folder | download | duplicates (3)
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')