diff --git a/t4_devkit/helper/rendering.py b/t4_devkit/helper/rendering.py index 2787e59..f1d739e 100644 --- a/t4_devkit/helper/rendering.py +++ b/t4_devkit/helper/rendering.py @@ -162,19 +162,23 @@ def render_scene( first_camera_tokens=first_camera_tokens, max_timestamp_us=max_timestamp_us, ) - ) - - # TODO(ktro2828): speed up annotation rendering - self._render_annotation3ds( - viewer=viewer, - first_sample_token=scene.first_sample_token, - max_timestamp_us=max_timestamp_us, - future_seconds=future_seconds, - ) - self._render_annotation2ds( - viewer=viewer, - first_sample_token=scene.first_sample_token, - max_timestamp_us=max_time_seconds, + + [ + self._executor.submit( + self._render_annotation3ds( + viewer=viewer, + first_sample_token=scene.first_sample_token, + max_timestamp_us=max_timestamp_us, + future_seconds=future_seconds, + ) + ), + self._executor.submit( + self._render_annotation2ds( + viewer=viewer, + first_sample_token=scene.first_sample_token, + max_timestamp_us=max_timestamp_us, + ) + ), + ] ) def render_instance( @@ -261,22 +265,26 @@ def render_instance( viewer=viewer, first_camera_tokens=first_camera_tokens, max_timestamp_us=max_timestamp_us, - ), - ) - - # TODO(ktro2828): speed up annotation rendering - self._render_annotation3ds( - viewer=viewer, - first_sample_token=first_sample.token, - max_timestamp_us=max_timestamp_us, - future_seconds=future_seconds, - instance_tokens=instance_tokens, - ) - self._render_annotation2ds( - viewer=viewer, - first_sample_token=first_sample.token, - max_timestamp_us=max_timestamp_us, - instance_tokens=instance_tokens, + ) + + [ + self._executor.submit( + self._render_annotation3ds( + viewer=viewer, + first_sample_token=first_sample.token, + max_timestamp_us=max_timestamp_us, + future_seconds=future_seconds, + instance_tokens=instance_tokens, + ) + ), + self._executor.submit( + self._render_annotation2ds( + viewer=viewer, + first_sample_token=first_sample.token, + max_timestamp_us=max_timestamp_us, + instance_tokens=instance_tokens, + ) + ), + ], ) def render_pointcloud(