【Omniverseの使い方】コードを使ってシーンを生成し、画像データセットを合成する

今回はコードを使用して物流倉庫を生成し、画像データセットを合成します

import os
import omni
from pxr import Usd, UsdGeom, Gf, UsdShade
from omni.isaac.synthetic_utils import SyntheticDataHelper

画像データセットを保存するためのパラメータを設定する

output_folder = "dataset/images"  # 图像数据集保存的文件夹路径
image_width = 640  # 图像宽度
image_height = 480  # 图像高度
num_images = 100  # 生成的图像数量

物流倉庫シーンの作成

def create_logistics_warehouse_scene():
    stage = omni.usd.get_context().get_stage()

# 创建物流仓库模型
warehouse_prim = stage.DefinePrim('/World/Warehouse', 'Xform')
warehouse_model = UsdGeom.Mesh.Define(stage, '/World/Warehouse/WarehouseModel')
# 设置物流仓库模型的位置、旋转等属性

# 添加其他物体、光照等场景元素
# ...

SyntheticDataHelper オブジェクトを初期化する

synthetic_helper = SyntheticDataHelper()
synthetic_helper.set_output_image_folder(output_folder)
synthetic_helper.set_output_image_size(image_width, image_height)

物流倉庫シーンの作成

create_logistics_warehouse_scene()

ループ内で画像データセットを生成する

for i in range(num_images):

# 渲染场景并获取图像

image = synthetic_helper.render_image()

# 保存图像到指定文件夹
image_path = os.path.join(output_folder, f"image_{i}.png")
image.save(image_path)
# 输出生成进度
print(f"Generated image {i+1}/{num_images}: {image_path}")

print("Image dataset generation complete!")

完全なコード:

```python
import os
import omni
from pxr import Usd, UsdGeom, Gf, UsdShade
from omni.isaac.synthetic_utils import SyntheticDataHelper

# 设置保存图像数据集的参数
output_folder = "dataset/images"  # 图像数据集保存的文件夹路径
image_width = 640  # 图像宽度
image_height = 480  # 图像高度
num_images = 100  # 生成的图像数量

# 创建物流仓库场景
def create_logistics_warehouse_scene():
    stage = omni.usd.get_context().get_stage()

    # 创建物流仓库模型
    warehouse_prim = stage.DefinePrim('/World/Warehouse', 'Xform')
    warehouse_model = UsdGeom.Mesh.Define(stage, '/World/Warehouse/WarehouseModel')
    # 设置物流仓库模型的位置、旋转等属性

    # 添加其他物体、光照等场景元素
    # ...

# 初始化SyntheticDataHelper对象
synthetic_helper = SyntheticDataHelper()
synthetic_helper.set_output_image_folder(output_folder)
synthetic_helper.set_output_image_size(image_width, image_height)

# 创建物流仓库场景
create_logistics_warehouse_scene()

# 循环生成图像数据集
for i in range(num_images):
    # 渲染场景并获取图像
    image = synthetic_helper.render_image()
    
    # 保存图像到指定文件夹
    image_path = os.path.join(output_folder, f"image_{i}.png")
    image.save(image_path)
    
    # 输出生成进度
    print(f"Generated image {i+1}/{num_images}: {image_path}")

print("Image dataset generation complete!")