diff --git a/README.md b/README.md index b3a1a6b02c..844c442fdc 100644 --- a/README.md +++ b/README.md @@ -272,7 +272,7 @@ To verify and configure GPU support for Docker, please follow the instructions p Once Docker is configured to use GPUs, you can run docTR Docker containers with GPU support: ```shell -docker run --it --gpus all ghcr.io/mindee/doctr:tf-py3.8.18-gpu-2023-09 bash +docker run -it --gpus all ghcr.io/mindee/doctr:tf-py3.8.18-gpu-2023-09 bash ``` #### Available Tags diff --git a/docs/source/modules/datasets.rst b/docs/source/modules/datasets.rst index 852ccd3841..94beaf6a2f 100644 --- a/docs/source/modules/datasets.rst +++ b/docs/source/modules/datasets.rst @@ -34,6 +34,8 @@ doctr.datasets .. autoclass:: DocArtefacts +.. autoclass:: WILDRECEIPT + Synthetic dataset generator --------------------------- diff --git a/doctr/datasets/wildreceipt.py b/doctr/datasets/wildreceipt.py index a802290110..496ddd87b2 100644 --- a/doctr/datasets/wildreceipt.py +++ b/doctr/datasets/wildreceipt.py @@ -21,6 +21,9 @@ class WILDRECEIPT(AbstractDataset): `_ | `repository `_. + .. image:: https://doctr-static.mindee.com/models?id=v0.7.0/wildreceipt-dataset.jpg&src=0 + :align: center + >>> # NOTE: You need to download the dataset first. >>> from doctr.datasets import WILDRECEIPT >>> train_set = WILDRECEIPT(train=True, img_folder="/path/to/wildreceipt/",