Initial commit with project files
This commit is contained in:
38
CV-App/algorithms/__init__.py
Normal file
38
CV-App/algorithms/__init__.py
Normal file
@@ -0,0 +1,38 @@
|
||||
|
||||
|
||||
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
|
||||
Reference in New Issue
Block a user