【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!")
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值