framework/PoseEstimation/BOPdataset.md

1.5 KiB
Raw Blame History

id title
BOP_dataset script for create BOP dataset

Структура входных данных:

<example_dir>/
    input_obj/asm_element_edge.mtl  # файл материала
    input_obj/asm_element_edge.obj  # меш-объект
    input_obj/fork.mtl
    input_obj/fork.obj
    input_obj/...
    resources/sklad.blend           # файл сцены
    objs2BOPdataset.py              # этот скрипт

Пример команды запуска скрипта:

cd <example_dir>/
blenderproc run objs2BOPdataset.py resources/sklad.blend input_obj output --imgs 333
  • resources/sklad.blend : файл сцены
  • input_obj : каталог с меш-файлами
  • output : выходной каталог
  • imgs : количество пакетов по 9 кадров в каждом (в примере 333 * 9 = 2997)

Структура BOP датасета на выходе:

output/
  bop_data/
    train_pbr/
      000000/
        depth/...          # файлы глубины
        mask/...           # файлы маски
        mask_visib/...     # файлы маски видимости
        rgb/...            # файлы изображений RGB
        scene_camera.json
        scene_gt.json
        scene_gt_coco.json
        scene_gt_info.json
    camera.json  # внутренние параметры камеры (для всего датасета)
  res.txt        # протокол создания пакетов датасета