39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
|
|
|
class Algorithm:
|
|
""" An abstract class to create custom algorithms """
|
|
def process(self, img):
|
|
""" Processes the input image"""
|
|
return img
|
|
|
|
def mouse_callback(self, event, x, y, flags, param):
|
|
""" Reacts on mouse callbacks """
|
|
return
|
|
|
|
|
|
''' Import algorithms to use'''
|
|
from .image_to_gray import ImageToGray
|
|
from .image_to_hue import ImageToHue
|
|
from .motion_detector import MotionDetector
|
|
from .white_balancing import WhiteBalancing
|
|
from .spin import Spin
|
|
from .segmentation_tracker import SegmentationTracker
|
|
#from .object_detection import ObjectDetector
|
|
#from .bottle_detection import BottleDetector
|
|
from .invis_cloak import InvisCloak
|
|
from .canny_edges import CannyEdgeDetector
|
|
|
|
''' Link Algorithms to keys '''
|
|
algorithms = dict()
|
|
algorithms["0"] = Algorithm
|
|
algorithms["1"] = ImageToGray
|
|
algorithms["2"] = ImageToHue
|
|
algorithms["3"] = MotionDetector
|
|
algorithms["4"] = WhiteBalancing
|
|
algorithms["5"] = Spin
|
|
algorithms["6"] = SegmentationTracker
|
|
algorithms["7"] = InvisCloak
|
|
#algorithms["7"] = ObjectDetector
|
|
#algorithms["8"] = BottleDetector
|
|
algorithms["9"] = CannyEdgeDetector
|