diff --git a/inference/conf/humanseg.yaml b/inference/conf/humanseg.yaml index 78195d9e8b4fe2d9ed2758ab76d8e5f88fd104a9..bdde20a172bf48c2f768473d0bb606ac578ac0cf 100644 --- a/inference/conf/humanseg.yaml +++ b/inference/conf/humanseg.yaml @@ -1,15 +1,14 @@ 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] + MEAN: [0.40787450980392154, 0.4575254901960784, 0.481078431372549] STD: [1.0, 1.0, 1.0] IMAGE_TYPE: "rgb" NUM_CLASSES: 2 CHANNELS : 3 PRE_PROCESSOR: "SegPreProcessor" PREDICTOR_MODE: "NATIVE" - BATCH_SIZE : 3 + BATCH_SIZE : 1