SSHAPE_Dataset
收藏SSHAPE 数据集生成器
概述
SSHAPE 数据集生成器是一个用于生成数据集的工具,主要依赖于 Blender 软件。该工具通过定义规则文件来控制数据集的生成,包括对象、诱饵、材质和颜色的设置。
安装
前提条件
- 需要安装 Blender 软件,推荐版本 >=2.91。
克隆仓库
通过以下命令克隆 SSHAPE 数据集生成器仓库: bash git clone https://github.com/M4tt3/SSHAPE_Dataset_generator.git
可选安装
安装 pip 和 tqdm
- pip:用于安装 tqdm 和其他 Python 库。
- tqdm:用于在渲染过程中显示进度条。
安装 pip: bash wget https://bootstrap.pypa.io/get-pip.py {BLENDER}/{BLENDER VERSION}/python/bin/python get-pip.py
安装 tqdm: bash {BLENDER}/{BLENDER VERSION}/python/bin/python -m pip install tqdm
设置规则
规则文件定义了数据集中的对象、诱饵、材质和颜色等。以下是一个规则文件的模板: json { "objects" : [], "decoys" : [], "materials" : [], "colors" : [], "macros" : [] }
形状
定义对象和诱饵,例如: json { "id" : 0, "name" : "pyramid", "file" : "pyramid.blend", "allowed_colors" : "all", "allowed_materials" : ["rubber", "metal"], "min_distance" : 0.87, "snap_to_plane" : true }
材质
定义材质及其属性,例如: json { "id" : 0, "name" : "matte", "file" : "matte.blend", "allowed_colors" : "all" }
颜色
定义颜色及其属性,例如: json { "name" : "white", "id" : 1, "hex" : "#ffffff", "opacity" : 1 }
宏
使用宏简化规则定义,例如: json { "objects" : [ { "allowed_colors" : "$warm_colors", ... } ... ], ... "macros" : { "warm_colors" : ["red", "orange", "yellow"] } }
文件结构
建议使用以下文件结构:
. └── My_dataset/ ├── data/ │ ├── objects/ │ ├── decoys/ │ ├── materials/ │ ├── base_scene.blend │ └── rules.json └── output/
可选配置文件
可以通过配置文件调整渲染参数,生成配置文件: bash blender --background --python build_config_file.py -- {OUTPUT FILE}
开始渲染
使用以下命令开始渲染: bash blender --background --python create_dataset.py -- {ARGUMENTS}
如果使用配置文件: bash blender --background --python create_dataset.py -- --config {PATH TO CONFIG}




