Make the 90 degree glb rotation optional (#347)

* Add option to not rotate 90 degrees for glb

* Update blender.py with the new option

Author: wuchangchang

kubric/core/objects.py

@@ -302,4 +302,10 @@ class FileBasedObject(PhysicalObject):
   # is used in a physics simulation.
   use_parenting_instead_of_join = tl.Bool(False)
 
+  # By default, the objects are rotated by 90 degrees around the X axis after
+  # loading. This is to reproduce the behavior of Blender UI. If this is set to
+  # True, the object will not be rotated by 90 degrees around the X axis after
+  # loading.
+  do_not_rotate_glb_90_degrees_after_import = tl.Bool(False)
+
   # TODO: trigger error when changing filenames or asset-id after the fact

kubric/renderer/blender.py

@@ -476,9 +476,10 @@ def _add_asset(self, obj: core.FileBasedObject):
                 bpy.ops.object.delete()
 
             assert len(bpy.context.selected_objects) == 1
-            blender_obj = bpy.context.selected_objects[0]
-            blender_obj.rotation_quaternion = (0.707107, -0.707107, 0, 0)
-            bpy.ops.object.transform_apply(location=False, rotation=True, scale=False)
+            if not obj.do_not_rotate_glb_90_degrees_after_import:
+              blender_obj = bpy.context.selected_objects[0]
+              blender_obj.rotation_quaternion = (0.707107, -0.707107, 0, 0)
+              bpy.ops.object.transform_apply(location=False, rotation=True, scale=False)
 
           elif extension == "fbx":
             bpy.ops.import_scene.fbx(filepath=obj.render_filename,