分割

UP支持分割任务训练、部署、推理的全部流程; 具体代码

配置文件

代码仓库其中包括常用算法配置文件

数据集相关模块

数据集类型包括:

cityscapes

数据集类型通过设置SegDataset的seg_type来选择，默认为cityscapes，配置文件示例如下:

dataset:
  type: cls
  kwargs:
    seg_type: cityscapes    # 默认为cityscapes，选项包括: [cityscapes]
    meta_file: cityscapes/fine_train.txt
    image_reader:
       type: fs_opencv
       kwargs:
         image_dir: cityscapes
         color_mode: RGB
    seg_label_reader:
       type: fs_opencv
       kwargs:
         image_dir: cityscapes
         color_mode: GRAY
    transformer: [*seg_rand_resize, *flip, *seg_crop_train, *to_tensor, *normalize]
    num_classes: *num_classes
    ignore_label: 255

部署模块

转换kestrel模型时，需要设置具体配置如下:

to_kestrel:
  toks_type: seg
  plugin: psyche
  model_name: model  # tar模型文件名的前缀以及meta.json中的model_name
  version: 1.0.0
  resize_hw: 640x1024