From 89f89442b99154e40424deac132b8f4d487cdd3e Mon Sep 17 00:00:00 2001 From: Bin Long Date: Thu, 26 Sep 2019 21:14:41 +0800 Subject: [PATCH] Update humanseg.yaml --- inference/conf/humanseg.yaml | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/inference/conf/humanseg.yaml b/inference/conf/humanseg.yaml index 78195d9e..bdde20a1 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 -- GitLab