fix boxlabel error

2022-08-16 03:34:40 +00:00 · 2022-08-16 03:34:40 +00:00 · ec22e60cb4
parent c1e6558d9e
commit ec22e60cb4
1 changed files with 1 additions and 1 deletions
--- a/ppocr/data/imaug/label_ops.py
+++ b/ppocr/data/imaug/label_ops.py
@ -750,7 +750,7 @@ class TableMasterLabelEncode(TableLabelEncode):

 class TableBoxEncode(object):
    def __init__(self, in_box_format='xyxy', out_box_format='xyxy', **kwargs):
-        assert box_format in ['xywh', 'xyxy', 'xyxyxyxy']
+        assert out_box_format in ['xywh', 'xyxy', 'xyxyxyxy']
        self.in_box_format = in_box_format
        self.out_box_format = out_box_format