add docu and tests

mindee · Nov 15, 2021 · 5d120c8 · 5d120c8
1 parent a1b581c
commit 5d120c8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/docs/source/datasets.rst b/docs/source/datasets.rst
@@ -20,6 +20,7 @@ Here are all datasets that are available through docTR:
 .. autoclass:: CharacterGenerator
 .. autoclass:: DocArtefacts
 .. autoclass:: IIIT5K
+.. autoclass:: SynthText
 
 
 Data Loading

diff --git a/tests/pytorch/test_datasets_pt.py b/tests/pytorch/test_datasets_pt.py
@@ -32,6 +32,8 @@ def test_visiondataset():
         ['DocArtefacts', False, [512, 512], 300, True],
         ['IIIT5K', True, [32, 128], 2000, True],
         ['IIIT5K', False, [32, 128], 3000, False],
+        ['SynthText', False, [512, 512], 858750, False],
+        ['SynthText', False, [512, 512], 858750, True],
     ],
 )
 def test_dataset(dataset_name, train, input_size, size, rotate):

diff --git a/tests/tensorflow/test_datasets_tf.py b/tests/tensorflow/test_datasets_tf.py
@@ -22,6 +22,8 @@
         ['DocArtefacts', False, [512, 512], 300, True],
         ['IIIT5K', True, [32, 128], 2000, True],
         ['IIIT5K', False, [32, 128], 3000, False],
+        ['SynthText', False, [512, 512], 858750, False],
+        ['SynthText', False, [512, 512], 858750, True],
     ],
 )
 def test_dataset(dataset_name, train, input_size, size, rotate):