Cube-Dataset-Preprocessing

Python script for preprocessing Cube dataset

Getting Started

Download the CubeDataset from here and extract the png files into a directory named images. Folder structure should look similar to the following:

- Cube-Dataset-Preprocessing/
                            data/
                                Cube/
                                    images/
                                          | 1.png
                                          | 2.png
                                          ...
                                    cube_gt.txt
                            CubeDataset.py
                            README.md
                            datasets.py

Run the following command:

To process a single image

python CubeDataset.py (index of image)
ex: python CubeDataset.py 143

To process multiple images ranging from n1 to n2

python CubeDataset.py 15 245

To dump processed images to pkl files (datasets incorporated from fc4)

python datasets.py