def to_proto( self, proto: Optional[image_util.ImagePreprocess] = None ) -> image_util.ImagePreprocess: proto = proto or image_util.ImagePreprocess() setattr(proto.resize, "width", self.width) setattr(proto.resize, "height", self.height) return proto
def to_proto( self, proto: Optional[image_util.ImagePreprocess] = None ) -> image_util.ImagePreprocess: if proto is None: proto = image_util.ImagePreprocess() if self.preprocessor == "bgr2rgb": proto.bgr2rgb.SetInParent() elif self.preprocessor == "mirror": proto.mirror.SetInParent() else: raise NotImplementedError return proto