modify pretrained model with training (#2505)

PaddlePaddle · Nov 21, 2024 · a81f88b · a81f88b
1 parent 415ce5c
commit a81f88b
Show file tree

Hide file tree

Showing 21 changed files with 21 additions and 21 deletions.
diff --git a/paddlex/configs/anomaly_detection/STFPM.yaml b/paddlex/configs/anomaly_detection/STFPM.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 1
   batch_size: 1
   learning_rate: 0.4
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/STFPM_pretrained.pdparams
+  pretrain_weight_path: null
   warmup_steps: 100
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_base.yaml b/paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_base.yaml
@@ -20,7 +20,7 @@ Train:
   epochs_iters: 20
   batch_size: 32
   learning_rate: 0.001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_base_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_base_backbone_LAION400M_pretrained.pdparams # use CLIP_vit_base_patch16_224 pretrained
   warmup_steps: 5
   resume_path: null
   log_interval: 1

diff --git a/paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_large.yaml b/paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_large.yaml
@@ -20,7 +20,7 @@ Train:
   epochs_iters: 20
   batch_size: 32
   learning_rate: 0.001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_large_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_large_backbone_LAION400M_pretrained.pdparams  # use CLIP_vit_large_patch14_224 pretrained
   warmup_steps: 5
   resume_path: null
   log_interval: 1

diff --git a/paddlex/configs/object_detection/PP-YOLOE_plus-L.yaml b/paddlex/configs/object_detection/PP-YOLOE_plus-L.yaml
@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-L_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-L_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/object_detection/PP-YOLOE_plus-M.yaml b/paddlex/configs/object_detection/PP-YOLOE_plus-M.yaml
@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-M_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-M_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/object_detection/PP-YOLOE_plus-S.yaml b/paddlex/configs/object_detection/PP-YOLOE_plus-S.yaml
@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-S_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-S_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/object_detection/PP-YOLOE_plus-X.yaml b/paddlex/configs/object_detection/PP-YOLOE_plus-X.yaml
@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-X_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-X_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/Deeplabv3-R101.yaml b/paddlex/configs/semantic_segmentation/Deeplabv3-R101.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R101_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R101_backbone_imagenet_pretrained.pdparams  # use ResNet101_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/Deeplabv3-R50.yaml b/paddlex/configs/semantic_segmentation/Deeplabv3-R50.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R50_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R50_backbone_imagenet_pretrained.pdparams  # use ResNet50_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R101.yaml b/paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R101.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R101_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R101_backbone_imagenet_pretrained.pdparams # use ResNet101_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R50.yaml b/paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R50.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R50_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R50_backbone_imagenet_pretrained.pdparams # use ResNet50_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/OCRNet_HRNet-W18.yaml b/paddlex/configs/semantic_segmentation/OCRNet_HRNet-W18.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W18_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W18_backbone_imagenet_pretrained.pdparams  # use HRNet_W18 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/OCRNet_HRNet-W48.yaml b/paddlex/configs/semantic_segmentation/OCRNet_HRNet-W48.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W48_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W48_backbone_imagenet_pretrained.pdparams  # use HRNet_W48 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/PP-LiteSeg-B.yaml b/paddlex/configs/semantic_segmentation/PP-LiteSeg-B.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-B_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-B_backbone_imagenet_pretrained.pdparams  # use STDC2 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/PP-LiteSeg-T.yaml b/paddlex/configs/semantic_segmentation/PP-LiteSeg-T.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-T_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-T_backbone_imagenet_pretrained.pdparams  # use STDC1 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/SegFormer-B0.yaml b/paddlex/configs/semantic_segmentation/SegFormer-B0.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B0_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B0_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B0 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/SegFormer-B1.yaml b/paddlex/configs/semantic_segmentation/SegFormer-B1.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B1_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B1_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B1 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/SegFormer-B2.yaml b/paddlex/configs/semantic_segmentation/SegFormer-B2.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B2_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B2_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B2 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/SegFormer-B3.yaml b/paddlex/configs/semantic_segmentation/SegFormer-B3.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B3_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B3_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B3 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/SegFormer-B4.yaml b/paddlex/configs/semantic_segmentation/SegFormer-B4.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B4_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B4_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B4 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

diff --git a/paddlex/configs/semantic_segmentation/SegFormer-B5.yaml b/paddlex/configs/semantic_segmentation/SegFormer-B5.yaml
@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B5_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B5_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B5 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10