gnes.preprocessor.image.segmentation module

class gnes.preprocessor.image.segmentation.SegmentPreprocessor(model_name: str, model_dir: str, _use_cuda: bool = False, *args, **kwargs)[source]

Bases: gnes.preprocessor.image.resize.SizedPreprocessor

apply(doc: gnes_pb2.Document)[source]
post_init()[source]

Declare class attributes/members that can not be serialized in standard way

train(*args, **kwargs)

Train the model, need to be overrided