Skip to content

Latest commit

 

History

History
37 lines (30 loc) · 1.45 KB

README.md

File metadata and controls

37 lines (30 loc) · 1.45 KB

Bring Your Own Data

Our model currently only support 360-degree data. Forward-Facing data is not supported for now.

Please install COLMAP before preparing your own data.

To begin with, organize your dataset as the following structure (see BYOD_example at here):

data\
|-- images_unmasked\
    |-- 1.jpg
    |-- 2.jpg
    |-- ...
|-- images_masks\
    |-- 1.jpg
    |-- 2.jpg
    |-- ...
|-- (optional) sparse\
    |-- 0\
        |-- images.bin
        |-- points3D.bin
        |-- cameras.bin

the image masks should be B/W masks. If your images are collected from different places, it is strongly recommended to mask out the image backgrounds with white color.

It is optional, however highly recommended, to execute COLMAP's SfM algorithm on its GUI by yourself. If your images are collected from the Internet, it might be very hard for COLMAP to solve a valid registration, and the GUI can be quite helpful for tunning.

It is also suggested to apply any advanced correspondence network, such as SuperGlue, to improve the quality of camera poses. See here.

After the dataset is prepared, run:

cd utils/data_preproccess
python 1_mask_image.py --scenedir <path of your dataset>
python 2_preprocess_llff.py --scenedir <path of your dataset>

and your data is ready to go.