humanseg.yaml

DEPLOY:
    USE_GPU: 1
    MODEL_PATH: "/root/projects/models/deeplabv3p_xception65_humanseg"
    MODEL_NAME: "unet"
    MODEL_FILENAME: "__model__"
    PARAMS_FILENAME: "__params__"
    EVAL_CROP_SIZE: (513, 513)
    MEAN: [0.5, 0.5, 0.5]
    STD: [1.0, 1.0, 1.0]
    IMAGE_TYPE: "rgb"
    NUM_CLASSES: 2
    CHANNELS : 3
    PRE_PROCESSOR: "SegPreProcessor"
    PREDICTOR_MODE: "NATIVE"
    BATCH_SIZE : 3