fix_server_v4_det_output (#14472)

PaddlePaddle · Jan 3, 2025 · 4f7476d · 4f7476d
1 parent a28196c
commit 4f7476d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/modeling/heads/det_db_head.py b/ppocr/modeling/heads/det_db_head.py
@@ -151,7 +151,7 @@ def forward(self, x, targets=None):
         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)