Update differentiable_renderer_tensorflow.py

martinResearch · Jul 15, 2023 · e8ea471 · e8ea471
1 parent d8b3187
commit e8ea471
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deodr/tensorflow/differentiable_renderer_tensorflow.py b/deodr/tensorflow/differentiable_renderer_tensorflow.py
@@ -68,7 +68,9 @@ def forward(
         scene.scene_2d.colors = np.array(colors)
 
         scene.scene_2d.depths = np.array(scene.scene_2d.depths)
-        differentiable_renderer_cython.renderSceneCpp(scene.scene_2d, 1, image, z_buffer)
+        differentiable_renderer_cython.renderSceneCpp(
+            scene.scene_2d, 1, image, z_buffer
+        )
 
         def backward(image_b: tf.Tensor) -> Tuple[tf.Tensor, tf.Tensor]:
             assert scene.scene_2d.colors is not None