GouMinghao / OCRTOC_dataset_toolkit

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.

Dataset Preparation

Download the dataset.

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 documentations for the toolkit. You may refer to the documentations for more details. The documentation is available at here.

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

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

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