yongsheng yuan 1 år sedan
förälder
incheckning
807ad1000d

+ 1 - 1
paddlex/configs/face_detection/PicoDet_LCNet_x2_5_face.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 1
   epochs_iters: 50
   batch_size: 4
   learning_rate: 0.08

+ 1 - 1
paddlex/configs/human_detection/PP-YOLOE-L_human.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 1
   epochs_iters: 50
   batch_size: 2
   learning_rate: 0.001

+ 1 - 1
paddlex/configs/human_detection/PP-YOLOE-S_human.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 1
   epochs_iters: 50
   batch_size: 2
   learning_rate: 0.001

+ 1 - 1
paddlex/configs/mainbody_detection/PP-ShiTuV2_det.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 1
   epochs_iters: 50
   batch_size: 2
   learning_rate: 0.4

+ 1 - 1
paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-L.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 10
   epochs_iters: 50
   batch_size: 2
   learning_rate: 0.01

+ 1 - 1
paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-S.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 10
   epochs_iters: 50
   batch_size: 2
   learning_rate: 0.01

+ 1 - 1
paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-largesize-L.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 10
   epochs_iters: 10
   batch_size: 2
   learning_rate: 0.00125