zoo.models.image.common package

Submodules

zoo.models.image.common.image_config module

class zoo.models.image.common.image_config.ImageConfigure(pre_processor=None, post_processor=None, batch_per_partition=4, label_map=None, feature_padding_param=None, jvalue=None, bigdl_type='float')

Bases: bigdl.util.common.JavaValue

predictor configure :param pre_processor preprocessor of ImageSet before model inference :param post_processor postprocessor of ImageSet after model inference :param batch_per_partition batch size per partition :param label_map mapping from prediction result indexes to real dataset labels :param feature_padding_param featurePaddingParam if the inputs have variant size

label_map()

class zoo.models.image.common.image_config.PaddingParam(bigdl_type='float')

Bases: bigdl.util.common.JavaValue

zoo.models.image.common.image_model module

class zoo.models.image.common.image_model.ImageModel(bigdl_type='float')

Bases: zoo.models.common.zoo_model.ZooModel

The basic class for image model.

get_config()

predict_image_set(image, configure=None)

Module contents