mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix dml (#6067)

pull/6113/head
Double_V 2022-05-01 19:54:22 +08:00 committed by GitHub
parent 77703f37fb
commit 6c68808a3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml
View File

@ -37,7 +37,7 @@ Architecture:
Head: Head:
name: DBHead name: DBHead
k: 50 k: 50
Student2: Teacher:
pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained
freeze_params: false freeze_params: false
return_all_feats: false return_all_feats: false
@ -62,15 +62,15 @@ Loss:
loss_config_list: loss_config_list:
- DistillationDMLLoss: - DistillationDMLLoss:
model_name_pairs: model_name_pairs:
- ["Student", "Student2"] - ["Student", "Teacher"]
maps_name: "thrink_maps" maps_name: "thrink_maps"
weight: 1.0 weight: 1.0
act: "softmax" act: "softmax"
model_name_pairs: ["Student", "Student2"] model_name_pairs: ["Student", "Teacher"]
key: maps key: maps
- DistillationDBLoss: - DistillationDBLoss:
weight: 1.0 weight: 1.0
model_name_list: ["Student", "Student2"] model_name_list: ["Student", "Teacher"]
# key: maps # key: maps
name: DBLoss name: DBLoss
balance_loss: true balance_loss: true
@ -94,7 +94,7 @@ Optimizer:
PostProcess: PostProcess:
name: DistillationDBPostProcess name: DistillationDBPostProcess
model_name: ["Student", "Student2"] model_name: ["Student", "Teacher"]
key: head_out key: head_out
thresh: 0.3 thresh: 0.3
box_thresh: 0.6 box_thresh: 0.6