OCRTOC Dataset Toolkit

Python toolkit for OCRTOC dataset.

Installation

git clone https://github.com/OCRTOC/OCRTOC_dataset_toolkit.git
cd OCRTOC_dataset_toolkit
pip install .
# python3 is supported, python2 is not tested.

Download the Dataset

The exact address for downloading is saved in the download.sh file. To download the dataset, you can run the following command.

bash download.sh YOUR_DATASET_ROOT

The dataset file structure is shown in structure.txt

Examples

Here are two examples on how to use this dataset toolkit.

cd examples

python example_loaddata.py --dataset_root YOUR_DATASET_ROOT

python example_vis_6dpose.py --dataset_root YOUR_DATASET_ROOT

Documentations

We also provide detailed documentation for the toolkit. You can refer to the documentation for more details. The documentation is available here.

You can also build the documentation locally by running the following command.

cd OCRTOC_dataset_toolkit/docs
pip install -r requirements.txt
bash build_doc.sh

The documentation will be generated in docs/build/html folder. You can open index.html in the browser to see the documentations.