Skip to content

A procedural blender pipeline to generate images for deep learning

License

Notifications You must be signed in to change notification settings

5trobl/BlenderProc

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BlenderProc

A procedural blender pipeline for image generation for deep learning.

Check out our arXiv paper (we are updating it from time to time): https://arxiv.org/abs/1911.01911

General

In general, one run of the pipeline first loads or constructs a 3D scene, then sets some camera positions inside this scene and renders different types of images (rgb, depth, normals etc.) for each of them. The blender pipeline consists of different modules, where each of them performs one step in the described process. The modules are selected, ordered and configured via a .yaml file.

To run the blender pipeline one just has to call the run.py script in the main directory together with the desired config file and any additional arguments.

python run.py config.yaml <additional arguments>

This runs all modules specified in the config file in a step-by-step fashion in the configured order.

Functionality

The following modules are already implemented and ready to use:

  • Loading: *.obj, SunCG, Replica scenes.
  • Lighting: Set, sample lights, automatic lighting of SunCG scenes.
  • Cameras: set, sample or load camera poses from file.
  • Rendering: RGB, depth, normal and segmentation images.
  • Merging: .hdf5 containers.

For advanced usage which is not covered by these modules, own modules can easily be implemented.

Examples

  • Basic scene: Basic example
  • Simple SUNCG scene: Loads a SUNCG scene and camera positions from file before rendering color, normal, segmentation and a depth images.
  • SUNCG scene with camera sampling: Loads a SUNCG scene and automatically samples camera poses in every room before rendering color, normal, segmentation and a depth images.
  • Replica dataset: Load a replica room, sample camera poses and render normal images.
  • COCO annotations: Write to a .json file containing COCO annotations for the objects in the scene.

... And much more!

First step

Now head on to the examples and check the README there: get some basic understanding of the config files, start exploring our examples and get a gist about writing yor own modules!

Change log

See our change log.

About

A procedural blender pipeline to generate images for deep learning

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%