Explorar o código

fix the num_classess of seaformer models to 150 for exporting

gaotingquan hai 1 ano
pai
achega
25f1843fcb

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SeaFormer_base.yaml

@@ -17,7 +17,7 @@ train_dataset:
   type: Dataset
   type: Dataset
   dataset_root: data/Cityscapes/
   dataset_root: data/Cityscapes/
   train_path: datasets/Cityscapes/train.txt
   train_path: datasets/Cityscapes/train.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: ResizeStepScaling
     - type: ResizeStepScaling
       min_scale_factor: 0.5
       min_scale_factor: 0.5
@@ -37,7 +37,7 @@ val_dataset:
   type: Dataset
   type: Dataset
   dataset_root: datasets/Cityscapes
   dataset_root: datasets/Cityscapes
   val_path: datasets/Cityscapes/val.txt
   val_path: datasets/Cityscapes/val.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: Normalize
     - type: Normalize
   mode: val
   mode: val

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SeaFormer_large.yaml

@@ -14,7 +14,7 @@ train_dataset:
   type: Dataset
   type: Dataset
   dataset_root: data/Cityscapes/
   dataset_root: data/Cityscapes/
   train_path: datasets/Cityscapes/train.txt
   train_path: datasets/Cityscapes/train.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: ResizeStepScaling
     - type: ResizeStepScaling
       min_scale_factor: 0.5
       min_scale_factor: 0.5
@@ -34,7 +34,7 @@ val_dataset:
   type: Dataset
   type: Dataset
   dataset_root: datasets/Cityscapes
   dataset_root: datasets/Cityscapes
   val_path: datasets/Cityscapes/val.txt
   val_path: datasets/Cityscapes/val.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: Normalize
     - type: Normalize
   mode: val
   mode: val

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SeaFormer_small.yaml

@@ -17,7 +17,7 @@ train_dataset:
   type: Dataset
   type: Dataset
   dataset_root: data/Cityscapes/
   dataset_root: data/Cityscapes/
   train_path: datasets/Cityscapes/train.txt
   train_path: datasets/Cityscapes/train.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: ResizeStepScaling
     - type: ResizeStepScaling
       min_scale_factor: 0.5
       min_scale_factor: 0.5
@@ -37,7 +37,7 @@ val_dataset:
   type: Dataset
   type: Dataset
   dataset_root: datasets/Cityscapes
   dataset_root: datasets/Cityscapes
   val_path: datasets/Cityscapes/val.txt
   val_path: datasets/Cityscapes/val.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: Normalize
     - type: Normalize
   mode: val
   mode: val

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SeaFormer_tiny.yaml

@@ -17,7 +17,7 @@ train_dataset:
   type: Dataset
   type: Dataset
   dataset_root: data/Cityscapes/
   dataset_root: data/Cityscapes/
   train_path: datasets/Cityscapes/train.txt
   train_path: datasets/Cityscapes/train.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: ResizeStepScaling
     - type: ResizeStepScaling
       min_scale_factor: 0.5
       min_scale_factor: 0.5
@@ -37,7 +37,7 @@ val_dataset:
   type: Dataset
   type: Dataset
   dataset_root: datasets/Cityscapes
   dataset_root: datasets/Cityscapes
   val_path: datasets/Cityscapes/val.txt
   val_path: datasets/Cityscapes/val.txt
-  num_classes: 19
+  num_classes: 150
   transforms:
   transforms:
     - type: Normalize
     - type: Normalize
   mode: val
   mode: val