add docu and tests

mindee · fg-mindee · Dec 10, 2021 · Nov 13, 2021 · Nov 15, 2021 · Nov 15, 2021
commit 6270c9335a6f91bc089738927815c71c68e53b42
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):