fix cls_x and bbox_x is possibly unbound (#10991)

PaddlePaddle · Sep 25, 2023 · 2751cb3 · 2751cb3
1 parent 8a52c99
commit 2751cb3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ppocr/modeling/heads/table_master_head.py b/ppocr/modeling/heads/table_master_head.py
@@ -90,11 +90,13 @@ def decode(self, input, feature, src_mask, tgt_mask):
             x = layer(x, feature, src_mask, tgt_mask)
 
         # cls head
+        cls_x = x
         for layer in self.cls_layer:
             cls_x = layer(x, feature, src_mask, tgt_mask)
         cls_x = self.norm(cls_x)
 
         # bbox head
+        bbox_x = x
         for layer in self.bbox_layer:
             bbox_x = layer(x, feature, src_mask, tgt_mask)
         bbox_x = self.norm(bbox_x)