Blenderpro虚拟数据集的生成

  • 简介
  • Blenderproc的安装环境配置
  • ....................................
  • 环境库配置
  • to be continue....

简介

这是第一次在csdn上写文章,一方面是为了记录自己的学习过程,方便复习回顾,另一方面在csdn上收获了很多,也希望能帮助到有需要的人。 基于项目需求,需要对Blenderproc进行学习,环境为ubuntu20.04.

github源码地址:https://github.com/DLR-RM/BlenderProc 论文原文:https://arxiv.org/abs/1911.01911 参考:

Blenderproc的安装环境配置

  1. 构建虚拟环境 为了方便日后的环境管理,我们在anaconda新建一个虚拟环境进行Blenderproc的安装,anaconda安装过程略,新环境命令如下:
conda create -n blender python = 3.7  # blender处为你的虚拟环境名称
conda activate blender
  1. 安装Blenderproc(这一步直接按github官方教程走) blenderproc的安装有两种方式,第一种是通过github安装,再这里我们使用更简便的方式,在虚拟环境下直接执行
pip install Blenderproc

安装结束后我们根据官方文档,先编写一个quickstart.py的python脚本,内容如下:

import blenderproc as bproc
import numpy as np

bproc.init()

# Create a simple object:
obj = bproc.object.create_primitive("MONKEY")

# Create a point light next to it
light = bproc.types.Light()
light.set_location([2, -2, 0])
light.set_energy(300)

# Set the camera to be in front of the object
cam_pose = bproc.math.build_transformation_mat([0, -5, 0], [np.pi / 2, 0, 0])
bproc.camera.add_camera_pose(cam_pose)

# Render the scene
data = bproc.renderer.render()

# Write the rendering into an hdf5 file
bproc.writer.write_hdf5("output/", data)

之后终端执行命令:

blenderproc run quickstart.py

第一次运行会安装blender和所需要的一些库,包括Imageio库(不用管),运行之后在目录下会有0.dhf5的文件生成,如图

blender生成stl Blender生成虚拟人_目标检测

此时终端运行:

blenderproc vis hdf5 output/0.hdf5

即可将结果进行可视化,结果如图:

blender生成stl Blender生成虚拟人_深度学习_02

到这一步,blenderproc的环境安装配置也就完成了,更多的基础示例可以下载github项目工程并参照示例进行。

3 注: blenderproc的运行有两种模式

  • 一种是通过设置python脚本运行(github项目工程主目录下运行)
blenderproc run examples/basics/basic/main.py examples/resources/camera_positions examples/resources/scene.obj examples/basics/basic/output
  • 另一种则是通过配置yaml文件进行运行
blenderproc run examples/basics/basic/config.yaml examples/resources/camera_positions examples/resources/scene.obj examples/basics/basic/output

两种方法大同小异,不过官方建议使用第一种方法,并且通过配置yaml文件运行的方法会在之后被移除。

环境库配置

在查看Blenderproc官方示例对于材质的改变时,我们发现缺少bpy这个库,但直接在虚拟环境中pip install bpy提示错误,安装失败。 参考 根据第一条内容,直接在whell文件发布地址按指引进行bpy安装。 whell文件发布地址:https://github.com/TylerGubala/blenderpy/releases

to be continue…