PP-HumanSegV2.yml 829 字节
Newer Older
W
wangguanzhong 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
ENV:
  min_subgraph_size: 3
  trt_calib_mode: False
  cpu_threads: 1
  trt_use_static: False
  save_img: True
  save_res: True
  return_res: True

MODEL:
  - SegmentationOp:
      name: seg
      param_path: paddlecv://models/PP_HumanSegV2_256x144_with_Softmax/model.pdiparams
      model_path: paddlecv://models/PP_HumanSegV2_256x144_with_Softmax/model.pdmodel
      batch_size: 8
      PreProcess:
        - Resize:
            target_size: [256, 144]
        - Normalize:
            scale: 0.00392157
            mean: [0.5, 0.5, 0.5]
            std: [0.5, 0.5, 0.5]
            order: ''
        - ToCHWImage
        - ExpandDim
      PostProcess:
        - SegPostProcess
      Inputs:
        - input.image

  - HumanSegOutput:
      name: out
      Inputs:
        - input.fn
        - input.image
        - seg.seg_map