Fix tabular printing of dataset information (#53)

Co-authored-by: linyiqi <linyq29@gmail.com>
2022-03-30 17:09:09 +08:00 · 2022-03-30 17:09:09 +08:00 · ef7116d77f
parent 8d6bf1d332
commit ef7116d77f
1 changed files with 2 additions and 0 deletions
--- a/mmfewshot/detection/datasets/base.py
+++ b/mmfewshot/detection/datasets/base.py
@ -539,6 +539,8 @@ class BaseFewShotDataset(CustomDataset):
            if len(row_data) == 10:
                table_data.append(row_data)
                row_data = []
+        if len(row_data) != 0:
+            table_data.append(row_data)

        table = AsciiTable(table_data)
        result += table.table