diff --git a/ppocr/modeling/heads/det_db_head.py b/ppocr/modeling/heads/det_db_head.py index 0100dbda0c..9f057b63ad 100644 --- 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)