[pre-commit.ci] pre-commit autoupdate

.pre-commit-config.yaml

@@ -11,7 +11,7 @@ repos:
  - id: trailing-whitespace
 
  - repo: https://github.com/astral-sh/ruff-pre-commit
- rev: v0.5.5
+ rev: v0.6.8
  hooks:
  - id: ruff
  types_or: [ python, pyi ]

demo.py

@@ -3,19 +3,20 @@
 
 
 def test_demo():
  import os
 
- import b3d
- import b3d.bayes3d as bayes3d
  import genjax
  import jax
  import jax.numpy as jnp
  import numpy as np
  import rerun as rr
- from b3d import Mesh, Pose
  from genjax import Pytree
  from tqdm import tqdm
 
+ import b3d
+ import b3d.bayes3d as bayes3d
+ from b3d import Mesh, Pose
+
  rr.init("demo")
  rr.connect("127.0.0.1:8812")
 

demos/detector_segmenter.py

@@ -1,25 +1,26 @@
 import io
 import os
 
-import b3d
 import jax
 import jax.numpy as jnp
 import numpy
 import numpy as np
 import rerun as rr
 import torch
 from PIL import Image
 from transformers import (
  AutoModelForZeroShotObjectDetection,
  AutoProcessor,
  DetrFeatureExtractor,
  DetrForSegmentation,
  OwlViTForObjectDetection,
  OwlViTProcessor,
 )
 from transformers.models.detr.feature_extraction_detr import rgb_to_id
 
+import b3d
+
 processor = OwlViTProcessor.from_pretrained("google/owlvit-base-patch32")
 model = OwlViTForObjectDetection.from_pretrained("google/owlvit-base-patch32")
 
 video_input = b3d.io.VideoInput.load(

demos/differentiable_renderer/dense_fitting/standalone_nishad_demo.py

@@ -1,16 +1,17 @@
 # Note: this uses a custom differentiable renderer defined inline in this script.
 
 import functools
 import os
 
-import b3d
 import jax
 import jax.numpy as jnp
 import numpy as np
 import rerun as rr
+
+import b3d
 from b3d import Pose
 
 rr.init("gradients")
 rr.connect("127.0.0.1:8812")
 
 # Load date

demos/differentiable_renderer/gradient_based_pose_estimation.py

@@ -1,16 +1,17 @@
 import os
 from functools import partial
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
 import jax
 import jax.numpy as jnp
 import optax
 import rerun as rr
-from b3d import Mesh, Pose
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
+from b3d import Mesh, Pose
+
 rr.init("gradients")
 rr.connect("127.0.0.1:8812")
 
 

demos/differentiable_renderer/gradients_for_mug.py

@@ -1,17 +1,18 @@
 import os
 from functools import partial
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
 import jax
 import jax.numpy as jnp
 import optax
 import rerun as rr
 import trimesh
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
+from b3d import Pose
+
 rr.init("gradients")
 rr.connect("127.0.0.1:8812")
 
 

demos/differentiable_renderer/patch_tracking/demo_utils.py

@@ -1,13 +1,14 @@
 import os
 
-import b3d
-import b3d.utils as utils
 import jax
 import jax.numpy as jnp
 import trimesh
+
+import b3d
+import b3d.utils as utils
 from b3d import Pose
 
 ### Utils ###
 
 
 # Define pose math libraries

demos/differentiable_renderer/patch_tracking/model.py

@@ -1,15 +1,15 @@
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
 import genjax
 import jax
 import jax.numpy as jnp
 import rerun as rr
-from b3d.modeling_utils import uniform_pose
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
 import demos.differentiable_renderer.patch_tracking.demo_utils as utils
+from b3d.modeling_utils import uniform_pose
 
 
 def normalize(v):
  return v / jnp.sum(v)
 
 

demos/differentiable_renderer/patch_tracking/multiple_patch_tracker_2.py

@@ -1,13 +1,13 @@
-import b3d
-import b3d.chisight.dense.patch_tracking as tracking
 import numpy as np
 import rerun as rr
-from b3d.chisight.dense.model import rr_log_uniformpose_meshes_to_image_model_trace
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.patch_tracking as tracking
 import demos.differentiable_renderer.patch_tracking.demo_utils as du
+from b3d.chisight.dense.model import rr_log_uniformpose_meshes_to_image_model_trace
 
 rr.init("multiple_patch_tracking_2")
 rr.connect("127.0.0.1:8812")
 
 width, height, fx, fy, cx, cy, near, far = 100, 128, 64.0, 64.0, 64.0, 64.0, 0.001, 16.0

demos/differentiable_renderer/patch_tracking/multiple_patch_tracking.py

@@ -1,24 +1,24 @@
 ### Preliminaries ###
 
 import os
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as r
-import b3d.chisight.dense.likelihoods as l
 import genjax
 import jax
 import jax.numpy as jnp
 import numpy as np
 import optax
 import rerun as rr
 import trimesh
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as r
+import b3d.chisight.dense.likelihoods as l
 import demos.differentiable_renderer.patch_tracking.demo_utils as du
 import demos.differentiable_renderer.patch_tracking.model as m
+from b3d import Pose
 
 rr.init("multiple_patch_tracking")
 rr.connect("127.0.0.1:8812")
 
 width = 100

demos/differentiable_renderer/patch_tracking/single_patch_tracking_adam.py

@@ -1,19 +1,19 @@
 ### Preliminaries ###
 
-import b3d.chisight.dense.differentiable_renderer as r
-import b3d.chisight.dense.likelihoods as l
 import genjax
 import jax
 import jax.numpy as jnp
 import optax
 import rerun as rr
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d.chisight.dense.differentiable_renderer as r
+import b3d.chisight.dense.likelihoods as l
 import demos.differentiable_renderer.patch_tracking.demo_utils as du
 import demos.differentiable_renderer.patch_tracking.model as m
+from b3d import Pose
 
 rr.init("single_patch_tracking")
 rr.connect("127.0.0.1:8812")
 
 (

demos/differentiable_renderer/patch_tracking/single_patch_tracking_mh.py

@@ -2,18 +2,18 @@
 
 import time
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as r
-import b3d.chisight.dense.likelihoods as l
 import genjax
 import jax
 import jax.numpy as jnp
 import rerun as rr
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as r
+import b3d.chisight.dense.likelihoods as l
 import demos.differentiable_renderer.patch_tracking.demo_utils as du
 import demos.differentiable_renderer.patch_tracking.model as m
+from b3d import Pose
 
 rr.init("single_patch_tracking-mh")
 rr.connect("127.0.0.1:8812")

demos/differentiable_renderer/test_barycentric_interp.py

@@ -1,9 +1,10 @@
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
-import b3d.chisight.dense.likelihoods as likelihoods
 import jax
 import jax.numpy as jnp
 import rerun as rr
+
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
+import b3d.chisight.dense.likelihoods as likelihoods
 from b3d import Pose
 
 # Set up OpenGL renderer

demos/differentiable_renderer/test_basic_gd.py

@@ -1,13 +1,13 @@
 import time
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
-import b3d.chisight.dense.likelihoods as likelihoods
 import genjax
 import jax
 import jax.numpy as jnp
 import rerun as rr
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
+import b3d.chisight.dense.likelihoods as likelihoods
 import demos.differentiable_renderer.utils as utils
 
 # Set up OpenGL renderer

demos/differentiable_renderer/utils.py

@@ -1,6 +1,7 @@
-import b3d
 import jax
 import jax.numpy as jnp
+
+import b3d
 from b3d import Pose
 
 

demos/fork_knife_smc_identity_pose.py

@@ -1,15 +1,16 @@
 import os
 
-import b3d
 import genjax
 import jax
 import jax.numpy as jnp
 import numpy as np
 import rerun as rr
 import trimesh
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d
+from b3d import Pose
+
 ### Choose experiment
 
 INPUT = "fork-visible" # TODO make one dataset with both objects?

demos/graphics_edits_demo/demo_visualize.py

@@ -2,15 +2,16 @@
 import os
 import pickle
 
-import b3d
 import jax
 import jax.numpy as jnp
 import numpy as np
 import rerun as rr
 import trimesh
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d
+from b3d import Pose
+
 rr.init("demo_visualize3")
 rr.connect("127.0.0.1:8812")
 

demos/graphics_edits_demo/vkm_demo.py

@@ -1,11 +1,12 @@
 import os
 
-import b3d
 import jax
 import jax.numpy as jnp
 import numpy as np
 import rerun as rr
 import trimesh
+
+import b3d
 from b3d import Pose
 
 rr.init("vkm_demo2")

demos/mesh_fitting/demo.py

@@ -1,13 +1,13 @@
 import os
 
-import b3d
 import genjax
 import jax
 import jax.numpy as jnp
 import optax
 import rerun as rr
 from tqdm import tqdm
 
+import b3d
 import demos.mesh_fitting.model as m
 import demos.mesh_fitting.tessellation as t
 

demos/mesh_fitting/demo_depth_init.py

@@ -1,10 +1,10 @@
 import os
 
-import b3d
 import genjax
 import jax
 import rerun as rr
 
+import b3d
 import demos.mesh_fitting.model as m
 import demos.mesh_fitting.utils as u
 

demos/mesh_fitting/model.py

@@ -1,9 +1,10 @@
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
 import genjax
 import jax
 import jax.numpy as jnp
 import rerun as rr
+
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
 from b3d.modeling_utils import uniform_pose
 
 

demos/mesh_fitting/utils.py

@@ -1,7 +1,7 @@
-import b3d
 import jax
 import jax.numpy as jnp
 
+import b3d
 import demos.mesh_fitting.tessellation as t
 
 

demos/posterior_datasets/identity_posterior_data_gen.py

@@ -1,12 +1,13 @@
 import os
 
-import b3d
 import jax
 import jax.numpy as jnp
 import numpy as np
 import rerun as rr
 import trimesh
 
+import b3d
+
 width = 200
 height = 200
 fx = 300.0

demos/posterior_datasets/pose_posterior_data_gen.py

@@ -1,10 +1,11 @@
 import os
 
-import b3d
 import jax
 import jax.numpy as jnp
 import rerun as rr
 import trimesh
+
+import b3d
 from b3d import Pose
 
 rr.init("demo")

demos/sparse_model/cotracker.py

@@ -1,10 +1,11 @@
 import argparse
 import time
 
-import b3d
 import numpy as np
 import torch
 
+import b3d
+
 parser = argparse.ArgumentParser("r3d_to_video_input")
 parser.add_argument("input", help=".r3d File", type=str)
 args = parser.parse_args()