Add a note about which tasks each dataset is for

robertknight · Feb 10, 2024 · 644c419 · 644c419
1 parent e2b30fe
commit 644c419
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ocrs_models/datasets.py b/ocrs_models/datasets.py
@@ -141,7 +141,7 @@ class SizedDataset(Dataset, Sized):
 
 class DDI100(SizedDataset):
     """
-    Distorted Document Images (DDI-100) dataset.
+    Distorted Document Images (DDI-100) dataset for text detection.
 
     See https://github.com/machine-intelligence-laboratory/DDI-100
     and https://arxiv.org/abs/1912.11658.
@@ -230,7 +230,7 @@ def reorder_quad(quad):
 
 class HierText(SizedDataset):
     """
-    HierText dataset.
+    HierText dataset for text detection.
 
     See https://github.com/google-research-datasets/hiertext and
     https://arxiv.org/abs/2203.15143.