Ver Fonte

adapt OCRNet & SegFormer (#3456)

Zhang Zelun há 8 meses atrás
pai
commit
c19cb542a0

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/OCRNet_HRNet-W18.yaml

@@ -1,5 +1,5 @@
 batch_size: 2
-iters: 80000
+iters: 160000
 
 train_dataset:
   type: Dataset
@@ -12,7 +12,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [512, 512]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/OCRNet_HRNet-W48.yaml

@@ -1,5 +1,5 @@
 batch_size: 2
-iters: 80000
+iters: 160000
 
 train_dataset:
   type: Dataset
@@ -12,7 +12,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [512, 512]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SegFormer-B0.yaml

@@ -1,5 +1,5 @@
 batch_size: 1
-iters: 80000
+iters: 160000
 
 model:
   type: SegFormer
@@ -19,7 +19,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [512, 512]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SegFormer-B1.yaml

@@ -1,5 +1,5 @@
 batch_size: 1
-iters: 80000
+iters: 160000
 
 model:
   type: SegFormer
@@ -19,7 +19,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [1024, 1024]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SegFormer-B2.yaml

@@ -1,5 +1,5 @@
 batch_size: 1
-iters: 80000
+iters: 160000
 
 model:
   type: SegFormer
@@ -19,7 +19,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [512, 512]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SegFormer-B3.yaml

@@ -1,5 +1,5 @@
 batch_size: 1
-iters: 80000
+iters: 160000
 
 model:
   type: SegFormer
@@ -19,7 +19,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [512, 512]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SegFormer-B4.yaml

@@ -1,5 +1,5 @@
 batch_size: 1
-iters: 80000
+iters: 160000
 
 model:
   type: SegFormer
@@ -19,7 +19,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [512, 512]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4

+ 2 - 2
paddlex/repo_apis/PaddleSeg_api/configs/SegFormer-B5.yaml

@@ -1,5 +1,5 @@
 batch_size: 1
-iters: 80000
+iters: 160000
 
 model:
   type: SegFormer
@@ -19,7 +19,7 @@ train_dataset:
       max_scale_factor: 2.0
       scale_step_size: 0.25
     - type: RandomPaddingCrop
-      crop_size: [512, 512]
+      crop_size: [1024, 512]
     - type: RandomHorizontalFlip
     - type: RandomDistort
       brightness_range: 0.4