Add a message when annotation is empty (#165)

Signed-off-by: lizz <lizz@sensetime.com>
2021-05-11 16:38:30 +08:00 · 2021-05-11 16:38:30 +08:00 · 6a1d1020f6
parent 85b821ab79
commit 6a1d1020f6
1 changed files with 1 additions and 0 deletions
--- a/mmocr/datasets/kie_dataset.py
+++ b/mmocr/datasets/kie_dataset.py
@ -80,6 +80,7 @@ class KIEDataset(BaseDataset):
        """

        assert utils.is_type_list(annotations, dict)
+        assert len(annotations) > 0, 'Please remove data with empty annotation'
        assert 'box' in annotations[0]
        assert 'text' in annotations[0]