diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py index 73d4ca430f6d3c234c640384d819a3c0dfa18470..59cb9b8a253cf04244ebf83511ab412174487a53 100644 --- 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