fix_server_v4_det_output (#14472)

2025-01-03 11:59:22 +08:00 · 2025-01-03 11:59:22 +08:00 · 4f7476d7b8
parent a28196c002
commit 4f7476d7b8
1 changed files with 1 additions and 1 deletions
--- a/ppocr/modeling/heads/det_db_head.py
+++ b/ppocr/modeling/heads/det_db_head.py
@ -151,7 +151,7 @@ class PFHeadLocal(DBHead):
        cbn_maps = self.cbn_layer(self.up_conv(f), shrink_maps, None)
        cbn_maps = F.sigmoid(cbn_maps)
        if not self.training:
-            return {"maps": 0.5 * (base_maps + cbn_maps), "cbn_maps": cbn_maps}
+            return {"maps": 0.5 * (base_maps + cbn_maps)}

        threshold_maps = self.thresh(x)
        binary_maps = self.step_function(shrink_maps, threshold_maps)