def __init__(self, pipeline_string=None):
if pipeline_string is None or not isinstance(pipeline_string, str):
raise FrameTransformationPipelineError("A 'pipeline_string' kwarg is expected...")
self.game_frame_transformer = FrameTransformer()
self.pipeline = self._parse_pipeline_string(pipeline_string)
self.pipeline_string = pipeline_string
def extract_game_area(self, frame_buffer):
game_area_buffer = []
for game_frame in frame_buffer.frames:
game_area = serpent.cv.extract_region_from_image(
game_frame.grayscale_frame,
self.game.screen_regions["GAME_REGION"])
frame = FrameTransformer.rescale(game_area, 0.25)
game_area_buffer.append(frame)
return game_area_buffer
def extract_game_area(self, frame_buffer):
game_area_buffer = []
for game_frame in frame_buffer.frames:
game_area = \
serpent.cv.extract_region_from_image(game_frame.grayscale_frame,self.game.screen_regions['GAME_REGION'])
frame = FrameTransformer.rescale(game_area, 0.25)
game_area_buffer.append(frame)
print(np.array(game_area_buffer).shape)
return np.array(game_area_buffer)