class Shwabe(object): def __init__(self): self.camera = Camera() self.processor = ImageProcessor() # self.mouse = Mouse(self.camera.shape[:2]) self.mouse = Mouse((480, 640)) self.brain = MainBrain() # self.mouse = Mouse(self.camera.get_camera_view()[:2]) def main_loop(self): while True: input_frame = self.camera.get_frame() # print("===========", input_frame.shape) self.mouse.update_view_size(input_frame.shape) # self.processor.extract_morph_from_img(input_frame) # center = self.processor.draw_circle() # if center: # self.mouse.move(center) # self.processor.draw_line() # self.processor.draw_windows() masked_image = self.processor.extract_mask(input_frame) self.brain.find_contours(masked_image) self.brain.find_defects_point() self.brain.draw_circle() # self.brain.draw_line() # self.brain.move_stats self.mouse.search_trigger(self.brain.move_stats, self.brain.move_cap, self.brain.movement_delta) self.brain.show_windows()