diff --git a/src/ptbench/data/loader.py b/src/ptbench/data/loader.py
index 2e03aa353c5f5e95db70b53072f94071b60cf677..0ee253a7ef8d5cb3106b8f958bcf872deb3e91bf 100644
--- a/src/ptbench/data/loader.py
+++ b/src/ptbench/data/loader.py
@@ -101,3 +101,36 @@ def make_delayed(sample, loader, key=None):
key=key or sample["data"],
label=sample["label"],
)
+
+def make_delayed_bbox(sample, loader, key=None):
+ """Returns a delayed-loading Sample object.
+
+ Parameters
+ ----------
+
+ sample : dict
+ A dictionary that maps field names to sample data values (e.g. paths)
+
+ loader : object
+ A function that inputs ``sample`` dictionaries and returns the loaded
+ data.
+
+ key : str
+ A unique key identifier for this sample. If not provided, assumes
+ ``sample`` is a dictionary with a ``data`` entry and uses its path as
+ key.
+
+
+ Returns
+ -------
+
+ sample : ptbench.data.sample.DelayedSample
+ In which ``key`` is as provided and ``data`` can be accessed to trigger
+ sample loading.
+ """
+ return DelayedSample(
+ functools.partial(loader, sample),
+ key=key or sample["data"],
+ label=sample["label"],
+ bboxes=sample["bboxes"],
+ )
diff --git a/src/ptbench/data/tbx11k_simplified/__init__.py b/src/ptbench/data/tbx11k_simplified/__init__.py
index e5910379ee8065c7f571a69a1bc60c2d3ef94e60..4652058176e7a63a724df9a8242ca5fd34adfb97 100644
--- a/src/ptbench/data/tbx11k_simplified/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified/__init__.py
@@ -28,7 +28,7 @@ import os
from ...utils.rc import load_rc
from ..dataset import JSONDataset
-from ..loader import load_pil_baw, make_delayed
+from ..loader import load_pil_baw, make_delayed, make_delayed_bbox
_protocols = [
importlib.resources.files(__name__).joinpath("default.json.bz2"),
@@ -48,23 +48,38 @@ _datadir = load_rc().get(
"datadir.tbx11k_simplified", os.path.realpath(os.curdir)
)
-
def _raw_data_loader(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
label=sample["label"],
)
+def _raw_data_loader_bbox(sample):
+ return dict(
+ data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
+ label=sample["label"],
+ bboxes=sample["bboxes"],
+ )
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader)
+def _loader_bbox(context, sample):
+ # "context" is ignored in this case - database is homogeneous
+ # we return delayed samples to avoid loading all images at once
+ return make_delayed_bbox(sample, _raw_data_loader_bbox)
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("data", "label"),
loader=_loader,
)
+
+dataset_with_bboxes = JSONDataset(
+ protocols=_protocols,
+ fieldnames=("data", "label", "bboxes"),
+ loader=_loader_bbox,
+)
"""TBX11K simplified dataset object."""
diff --git a/src/ptbench/data/tbx11k_simplified/default.json.bz2 b/src/ptbench/data/tbx11k_simplified/default.json.bz2
index e24f2b37061451cbbee4a322739189cd1fa4affc..06e5a3a7011f976118ae1067785e59b9e294480c 100644
Binary files a/src/ptbench/data/tbx11k_simplified/default.json.bz2 and b/src/ptbench/data/tbx11k_simplified/default.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_0.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_0.json.bz2
index 136016cac5823e3016c5e5ed04263415a242ffd2..267555153aebab7b1a0cfdd3dca0d4e7149cf5f3 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_0.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_0.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_1.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_1.json.bz2
index 9e3a4c621534e2709fcf4ae1b6239ce7afa66007..ceaa92a210722a839b40852d6dbb866b09712a99 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_1.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_1.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_2.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_2.json.bz2
index a7b3de7bd1ad85cb7fa51c3f7c4db24175ab7690..615ecfe9ddb562c27b14fcefd531b131a337f1b4 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_2.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_2.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_3.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_3.json.bz2
index 674167aa64080a5594bafe170325d6aac3cd394a..aee0fea3dc3095b6fb703b33365248a0439888d3 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_3.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_3.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_4.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_4.json.bz2
index ef064820c6bf5d9a112b206f4cd4eaed0d6d4f1d..a312ef4cb65aad35644b0e862da9be9ffce8f44f 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_4.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_4.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_5.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_5.json.bz2
index d2d969be9beacd6b90ed4d3deb00e9708a64b67d..926d83ed57633fe7388f744d904723821ece7bd7 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_5.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_5.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_6.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_6.json.bz2
index b71b5e43fa97303092c79c8fb2a0622c6d20fb40..47b8d130e7523fa9ab4a4f8927e2302dbbaeb352 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_6.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_6.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_7.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_7.json.bz2
index 65c5d5a3197b4ebeaee78adc958de15c72534225..7300fee208aff942afd5cc25845ecb7b3c55f859 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_7.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_7.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_8.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_8.json.bz2
index f2f733cad9be90235dee0359c423048504a63d2e..b7056757fa5af3fccb19eb03d52f0bcd72d13571 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_8.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_8.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified/fold_9.json.bz2 b/src/ptbench/data/tbx11k_simplified/fold_9.json.bz2
index 7af59bdea4f3d2a546052c4f21e0e4f7cada89fb..ef2c46e58692ba23816fcd3c853ae4a702acd591 100644
Binary files a/src/ptbench/data/tbx11k_simplified/fold_9.json.bz2 and b/src/ptbench/data/tbx11k_simplified/fold_9.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_RS/__init__.py b/src/ptbench/data/tbx11k_simplified_RS/__init__.py
index 8e782d45d55600f9af1d23fcbe907f0e2e437a71..1f328703dbb13e77983e761c2dad35c7aac276c2 100644
--- a/src/ptbench/data/tbx11k_simplified_RS/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_RS/__init__.py
@@ -27,7 +27,7 @@ The version of the dataset used in this benchmark is a simplified.
import importlib.resources
from ..dataset import JSONDataset
-from ..loader import make_delayed
+from ..loader import make_delayed, make_delayed_bbox
_protocols = [
importlib.resources.files(__name__).joinpath("default.json.bz2"),
@@ -47,16 +47,33 @@ _protocols = [
def _raw_data_loader(sample):
return dict(data=sample["data"], label=sample["label"])
+def _raw_data_loader_bbox(sample):
+ return dict(
+ data=sample["data"],
+ label=sample["label"],
+ bboxes=sample["bboxes"],
+ )
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we returned delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader, key=sample["filename"])
+def _loader_bbox(context, sample):
+ # "context" is ignored in this case - database is homogeneous
+ # we return delayed samples to avoid loading all images at once
+ return make_delayed_bbox(sample, _raw_data_loader_bbox, key=sample["filename"])
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("filename", "label", "data"),
loader=_loader,
)
+
+dataset_with_bboxes = JSONDataset(
+ protocols=_protocols,
+ fieldnames=("filename", "label", "data", "bboxes"),
+ loader=_loader_bbox,
+)
+
"""Extended TBX11K simplified dataset object."""
diff --git a/src/ptbench/data/tbx11k_simplified_RS_v2/__init__.py b/src/ptbench/data/tbx11k_simplified_RS_v2/__init__.py
index bec7223465a8b7265c497d42d604b95cfef68457..3a61ae3b2cffdb51cd9f9c52f17524eaff6a2a5d 100644
--- a/src/ptbench/data/tbx11k_simplified_RS_v2/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_RS_v2/__init__.py
@@ -47,16 +47,33 @@ _protocols = [
def _raw_data_loader(sample):
return dict(data=sample["data"], label=sample["label"])
+def _raw_data_loader_bbox(sample):
+ return dict(
+ data=sample["data"],
+ label=sample["label"],
+ bboxes=sample["bboxes"],
+ )
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we returned delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader, key=sample["filename"])
+def _loader_bbox(context, sample):
+ # "context" is ignored in this case - database is homogeneous
+ # we return delayed samples to avoid loading all images at once
+ return make_delayed_bbox(sample, _raw_data_loader_bbox, key=sample["filename"])
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("filename", "label", "data"),
loader=_loader,
)
+
+dataset_with_bboxes = JSONDataset(
+ protocols=_protocols,
+ fieldnames=("filename", "label", "data", "bboxes"),
+ loader=_loader_bbox,
+)
+
"""Extended TBX11K simplified dataset object."""
diff --git a/src/ptbench/data/tbx11k_simplified_RS_v3/__init__.py b/src/ptbench/data/tbx11k_simplified_RS_v3/__init__.py
index bec7223465a8b7265c497d42d604b95cfef68457..3a61ae3b2cffdb51cd9f9c52f17524eaff6a2a5d 100644
--- a/src/ptbench/data/tbx11k_simplified_RS_v3/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_RS_v3/__init__.py
@@ -47,16 +47,33 @@ _protocols = [
def _raw_data_loader(sample):
return dict(data=sample["data"], label=sample["label"])
+def _raw_data_loader_bbox(sample):
+ return dict(
+ data=sample["data"],
+ label=sample["label"],
+ bboxes=sample["bboxes"],
+ )
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we returned delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader, key=sample["filename"])
+def _loader_bbox(context, sample):
+ # "context" is ignored in this case - database is homogeneous
+ # we return delayed samples to avoid loading all images at once
+ return make_delayed_bbox(sample, _raw_data_loader_bbox, key=sample["filename"])
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("filename", "label", "data"),
loader=_loader,
)
+
+dataset_with_bboxes = JSONDataset(
+ protocols=_protocols,
+ fieldnames=("filename", "label", "data", "bboxes"),
+ loader=_loader_bbox,
+)
+
"""Extended TBX11K simplified dataset object."""
diff --git a/src/ptbench/data/tbx11k_simplified_v2/__init__.py b/src/ptbench/data/tbx11k_simplified_v2/__init__.py
index bf06074dc67890e9b5e843fac543964b9f032778..9cb5b281b0646060f8c614b7f3e012e1d7d0bc36 100644
--- a/src/ptbench/data/tbx11k_simplified_v2/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_v2/__init__.py
@@ -28,7 +28,7 @@ import os
from ...utils.rc import load_rc
from ..dataset import JSONDataset
-from ..loader import load_pil_baw, make_delayed
+from ..loader import load_pil_baw, make_delayed, make_delayed_bbox
_protocols = [
importlib.resources.files(__name__).joinpath("default.json.bz2"),
@@ -45,26 +45,41 @@ _protocols = [
]
_datadir = load_rc().get(
- "datadir.tbx11k_simplified_v2", os.path.realpath(os.curdir)
+ "datadir.tbx11k_simplified", os.path.realpath(os.curdir)
)
-
def _raw_data_loader(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
label=sample["label"],
)
+def _raw_data_loader_bbox(sample):
+ return dict(
+ data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
+ label=sample["label"],
+ bboxes=sample["bboxes"],
+ )
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader)
+def _loader_bbox(context, sample):
+ # "context" is ignored in this case - database is homogeneous
+ # we return delayed samples to avoid loading all images at once
+ return make_delayed_bbox(sample, _raw_data_loader_bbox)
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("data", "label"),
loader=_loader,
)
+
+dataset_with_bboxes = JSONDataset(
+ protocols=_protocols,
+ fieldnames=("data", "label", "bboxes"),
+ loader=_loader_bbox,
+)
"""TBX11K simplified dataset object."""
diff --git a/src/ptbench/data/tbx11k_simplified_v2/default.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/default.json.bz2
index cd9e57461b0e8a1d0a6193dec8b77b20f821a710..1947e1ec8b6e45c82ed467367792151f3a06def3 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/default.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/default.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_0.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_0.json.bz2
index 99382ab8bfc0e5acb2be9dfe3254f64889582074..beff51ae634b349f40e955ff8739385f51b7b067 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_0.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_0.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_1.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_1.json.bz2
index ebc0dfbeb2e03aa6376fcebf7485c0164fc079de..ed2f98d133fd97135830c6f20e1e0114e1264992 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_1.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_1.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_2.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_2.json.bz2
index af47edcf785e90e0ee6444d53cd43916f2947b1b..f3ab3352a2c3a3f613f084c77458ee36555d52b8 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_2.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_2.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_3.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_3.json.bz2
index b88660ec81dc1b2a9957abe129b2e5608c78bbca..f6e0c5c17e74e9171989f6e4fd0185047a78fe36 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_3.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_3.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_4.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_4.json.bz2
index 6302288ad8bfa625edfbd99c892db9ae86644b53..473793dd2bd8b0830132f54c16ff2458871122d6 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_4.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_4.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_5.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_5.json.bz2
index 1968ddc26dbd4d0797cc7b93fe811aadcf465004..40c492e496a4951a0aee98702b87affbec7d0af8 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_5.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_5.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_6.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_6.json.bz2
index 3693f6abb4c02c94e8d8116f471763da43b8746a..3324f8d899422800184777407159ba38ae0088d4 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_6.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_6.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_7.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_7.json.bz2
index e3ee6508b90744c604d65d93d99f47bbdb13bd05..6a435bced6808f41476d8635484d9b3e7356b5ef 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_7.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_7.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_8.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_8.json.bz2
index 97e5cd48f4a32f4312a772c985a98c53d4f5ff27..126734afadde81f1d861043b63d331f8b918548a 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_8.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_8.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v2/fold_9.json.bz2 b/src/ptbench/data/tbx11k_simplified_v2/fold_9.json.bz2
index e4e65b2b35342880157d8de39fb7b0b8ba1494ea..cde889ebdeb3bef2bc8b5877447140003300969d 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v2/fold_9.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v2/fold_9.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/__init__.py b/src/ptbench/data/tbx11k_simplified_v3/__init__.py
index e6adab57a7dcce33f089339c24ed50fdc9ce2eea..9cb5b281b0646060f8c614b7f3e012e1d7d0bc36 100644
--- a/src/ptbench/data/tbx11k_simplified_v3/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_v3/__init__.py
@@ -28,7 +28,7 @@ import os
from ...utils.rc import load_rc
from ..dataset import JSONDataset
-from ..loader import load_pil_baw, make_delayed
+from ..loader import load_pil_baw, make_delayed, make_delayed_bbox
_protocols = [
importlib.resources.files(__name__).joinpath("default.json.bz2"),
@@ -45,26 +45,41 @@ _protocols = [
]
_datadir = load_rc().get(
- "datadir.tbx11k_simplified_v3", os.path.realpath(os.curdir)
+ "datadir.tbx11k_simplified", os.path.realpath(os.curdir)
)
-
def _raw_data_loader(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
label=sample["label"],
)
+def _raw_data_loader_bbox(sample):
+ return dict(
+ data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
+ label=sample["label"],
+ bboxes=sample["bboxes"],
+ )
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader)
+def _loader_bbox(context, sample):
+ # "context" is ignored in this case - database is homogeneous
+ # we return delayed samples to avoid loading all images at once
+ return make_delayed_bbox(sample, _raw_data_loader_bbox)
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("data", "label"),
loader=_loader,
)
+
+dataset_with_bboxes = JSONDataset(
+ protocols=_protocols,
+ fieldnames=("data", "label", "bboxes"),
+ loader=_loader_bbox,
+)
"""TBX11K simplified dataset object."""
diff --git a/src/ptbench/data/tbx11k_simplified_v3/default.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/default.json.bz2
index b4aaee728338da92892e49520b62d1d28782e46a..53b1c72281098faa727ce9b578210ae588e204c0 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/default.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/default.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_0.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_0.json.bz2
index 22c8fc0fc32dcaf887a0a69a356a3720a35e8603..881297d6973db88dc2d6d1e19eb0c29f9062ed8a 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_0.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_0.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_1.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_1.json.bz2
index 886def48ad140e1341204592a4819ffd694aba23..d85439be416a48142310799e6bf32b286224dd6a 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_1.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_1.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_2.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_2.json.bz2
index e56ff0b6d0eec32ae519a179b88f5228d5cf66f3..fb8c89af52076546d2a4da4831eb5f650c93cf18 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_2.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_2.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_3.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_3.json.bz2
index 89d9e8dfc673ceecfa3609eae80287383bd58a26..0a8cc88fd74d7847785ea4a74cd7ced8c72fa89f 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_3.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_3.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_4.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_4.json.bz2
index ab832871e17ce89c88c3e06eb86157cb6be1f943..1ab03dfa21c6d8fe60a910343c4544f98e723634 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_4.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_4.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_5.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_5.json.bz2
index af2fe3d5a57139baee1c1b900cc9faf06eed40f4..17467ae2146adc756ebe7843e234aea18c5b9b14 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_5.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_5.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_6.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_6.json.bz2
index 240e55cf7ead833c4f65061680a65482bfaaadde..2359d5e1ee6f14491715842f81a244acd083f7da 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_6.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_6.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_7.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_7.json.bz2
index f315b5ebcb5ecaaea879d6de964643c0220a58f6..4cd04d5fd74a65704de41da44b5e638fa609a7fb 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_7.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_7.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_8.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_8.json.bz2
index 3e971e1b0a0a24d3931410975da6ff4e369007a8..fd8e62f18bf68ef535dc8e13f1291f9bdd5ad800 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_8.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_8.json.bz2 differ
diff --git a/src/ptbench/data/tbx11k_simplified_v3/fold_9.json.bz2 b/src/ptbench/data/tbx11k_simplified_v3/fold_9.json.bz2
index f481a22dbb7ccb820e328e19812c1f54681657c4..7018d30fae5a4a50f7aca57961274c3e0ef30987 100644
Binary files a/src/ptbench/data/tbx11k_simplified_v3/fold_9.json.bz2 and b/src/ptbench/data/tbx11k_simplified_v3/fold_9.json.bz2 differ