framework/simulation/pose_estimation/BOPdataset.md

45 lines
1.5 KiB
Markdown
Raw Normal View History

---
id: BOP_dataset
title: 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 # протокол создания пакетов датасета
```