class orb_brute(AbstractMethod):
orb: ORBBrute
def __init__(self):
self.orb = ORBBrute()
def query(self, picture: Picture) -> List[Picture]:
res = self.orb.query(picture)
return res
def train(self, images: List[Picture]):
self.orb.train(images)
class w5_no_frame_no_text(AbstractMethod):
orb: ORBBrute
def __init__(self):
self.orb = ORBBrute()
def query(self, picture: Picture) -> (List[Picture], Frame):
frame = get_frame_with_lines(picture.get_image())
return self.orb.query(picture), frame
def train(self, images: List[Picture]) -> List[Rectangle]:
return self.orb.train(images, use_mask=False)
def __init__(self):
self.orb = ORBBrute()