def post_process(self, img, size=None, swap=True): if img is not None: if size: img = resize(img, *size) if swap: img = swap_rb(img) return img
def get_image_data(self, size=None): if self._sock is None: img = self._get_image_data() else: img = self._get_video_data() if img is not None: if size: img = resize(img, *size) return asarray(img[:, :])