[datasets] Fix loading targets in WILDRECEIPT init (mindee#1859)

felixT2K · Jan 30, 2025 · 1c9ce92 · 1c9ce92
1 parent bdb2c28
commit 1c9ce92
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/doctr/datasets/wildreceipt.py b/doctr/datasets/wildreceipt.py
@@ -79,11 +79,11 @@ def __init__(
         # Split the text file into separate JSON strings
         json_strings = data.strip().split("\n")
         box: list[float] | np.ndarray
-        _targets = []
 
         for json_string in tqdm(
             iterable=json_strings, desc="Preparing and Loading WILDRECEIPT", total=len(json_strings)
         ):
+            _targets = []
             json_data = json.loads(json_string)
             img_path = json_data["file_name"]
             annotations = json_data["annotations"]

diff --git a/tests/pytorch/test_datasets_pt.py b/tests/pytorch/test_datasets_pt.py
@@ -734,7 +734,7 @@ def test_ic03(input_size, num_samples, rotate, recognition, detection, mock_ic03
     "input_size, num_samples, recognition, detection",
     [
         [[512, 512], 2, False, False],  # Actual set has 1268 training samples and 472 test samples
-        [[32, 128], 5, True, False],  # recognition
+        [[32, 128], 3, True, False],  # recognition
         [[512, 512], 2, False, True],  # detection
     ],
 )

diff --git a/tests/tensorflow/test_datasets_tf.py b/tests/tensorflow/test_datasets_tf.py
@@ -707,7 +707,7 @@ def test_ic03(input_size, num_samples, rotate, recognition, detection, mock_ic03
     "input_size, num_samples, recognition, detection",
     [
         [[512, 512], 2, False, False],  # Actual set has 1268 training samples and 472 test samples
-        [[32, 128], 5, True, False],  # recognition
+        [[32, 128], 3, True, False],  # recognition
         [[512, 512], 2, False, True],  # detection
     ],
 )