def _feed(self, data): color = data[1] if len(data) > 1 else self.color_space data = data[0] bbox = ImageLoader.get_image_bbox(self, None, data.shape[:2]) self.minibatch_data.mem[0], _, _ = self.preprocess_image( data, color, True, bbox)
def _feed(self, data, request): color = request.get("color_space", self.color_space) bbox = ImageLoader.get_image_bbox(self, None, data.shape[:2]) self.minibatch_data.mem[self._minibatch_size_], _, _ = \ self.preprocess_image(data, color, True, bbox)