Selaa lähdekoodia

fix layput num_class (#3379)

Sunflower7788 9 kuukautta sitten
vanhempi
commit
fb78883a2d

+ 1 - 1
paddlex/repo_apis/PaddleDetection_api/configs/PP-DocLayout-L.yaml

@@ -18,7 +18,7 @@ eval_size: [640, 640]
 
 # Dataset
 metric: COCO
-num_classes: 80
+num_classes: 23
 
 worker_num: 4
 

+ 1 - 1
paddlex/repo_apis/PaddleDetection_api/configs/PP-DocLayout-M.yaml

@@ -14,7 +14,7 @@ print_params: false
 
 # Dataset
 metric: COCO
-num_classes: 17
+num_classes: 23
 
 worker_num: 6
 eval_height: &eval_height 640

+ 1 - 1
paddlex/repo_apis/PaddleDetection_api/configs/PP-DocLayout-S.yaml

@@ -14,7 +14,7 @@ print_params: false
 
 # Dataset
 metric: COCO
-num_classes: 17
+num_classes: 23
 
 worker_num: 6
 eval_height: &eval_height 480