diff --git a/dan/manager/training.py b/dan/manager/training.py index c52cebd68b21b156df6690e1e0ff91f9cf5916cb..df33eda3b58bd1e9ba2c81ef966f2343c3cc5e0d 100644 --- a/dan/manager/training.py +++ b/dan/manager/training.py @@ -1042,7 +1042,6 @@ class OCRManager(GenericTrainingManager): { "path": sample["path"], "label": chunk, - "nb_cols": 1, } ) @@ -1055,7 +1054,6 @@ class OCRManager(GenericTrainingManager): Image.fromarray(img).save(img_path) gt[set_name][img_name] = { "text": sample["label"], - "nb_cols": sample["nb_cols"] if "nb_cols" in sample else 1, } if "line_label" in sample: gt[set_name][img_name]["lines"] = sample["line_label"]