要实现3个语义分割目标,可以在模型配置文件中将num_classes设置为3。例如,如果使用DeepLabV3+模型,则可以在配置文件中添加以下代码:

backbone:
  type: ResNet50_vd
  output_stride: 16

model:
  type: DeepLabV3P
  num_classes: 3

对应标签数据需要修改,需要将每个像素的标签值从原来的单一类别改为表示3个目标的多类别标签。例如,如果原来的标签值是0和1,表示两个类别,现在需要添加一个类别,可以将其中一个类别的标签值修改为2。这样,就可以将原来的两个类别表示为0和1,新添加的类别表示为2。例如,如果使用PaddleSeg自带的Cityscapes数据集,可以在标签文件中将原来的标签值修改为:

0 -> 0 (road)
1 -> 1 (sidewalk)
2 -> 2 (building)
3 -> 2 (wall)
4 -> 2 (fence)
5 -> 3 (pole)
6 -> 4 (traffic light)
7 -> 5 (traffic sign)
8 -> 6 (vegetation)
9 -> 7 (terrain)
10 -> 8 (sky)
11 -> 9 (person)
12 -> 10 (rider)
13 -> 11 (car)
14 -> 12 (truck)
15 -> 13 (bus)
16 -> 14 (train)
17 -> 15 (motorcycle)
18 -> 16 (bicycle)

这样,就可以将原来的19个类别表示为0到18,新添加的2个类别表示为19和20。

paddleseg的ppliteseg中想要有3个语义分割目标num_class怎么办对应标签数据如何给

原文地址: http://www.cveoy.top/t/topic/bE9T 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录