Skip to content

Commit 2ce515f

Browse files
eightysteeleeightysteele
committed
fix python errors
1 parent 8e81b6d commit 2ce515f

File tree

88 files changed

+601
-587
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

88 files changed

+601
-587
lines changed

‎demos/detector_segmenter.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import io
22
import os
33

4+
import b3d
45
import jax
56
import jax.numpy as jnp
67
import numpy
@@ -18,8 +19,6 @@
1819
)
1920
from transformers.models.detr.feature_extraction_detr import rgb_to_id
2021

21-
import b3d
22-
2322
processor = OwlViTProcessor.from_pretrained("google/owlvit-base-patch32")
2423
model = OwlViTForObjectDetection.from_pretrained("google/owlvit-base-patch32")
2524

‎demos/differentiable_renderer/dense_fitting/standalone_nishad_demo.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,11 @@
33
import functools
44
import os
55

6+
import b3d
67
import jax
78
import jax.numpy as jnp
89
import numpy as np
910
import rerun as rr
10-
11-
import b3d
1211
from b3d import Pose
1312

1413
rr.init("gradients")

‎demos/differentiable_renderer/gradient_based_pose_estimation.py‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,14 @@
11
import os
22
from functools import partial
33

4+
import b3d
5+
import b3d.chisight.dense.differentiable_renderer as rendering
46
import jax
57
import jax.numpy as jnp
68
import optax
79
import rerun as rr
8-
from tqdm import tqdm
9-
10-
import b3d
11-
import b3d.chisight.dense.differentiable_renderer as rendering
1210
from b3d import Mesh, Pose
11+
from tqdm import tqdm
1312

1413
rr.init("gradients")
1514
rr.connect("127.0.0.1:8812")

‎demos/differentiable_renderer/gradients_for_mug.py‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,15 @@
11
import os
22
from functools import partial
33

4+
import b3d
5+
import b3d.chisight.dense.differentiable_renderer as rendering
46
import jax
57
import jax.numpy as jnp
68
import optax
79
import rerun as rr
810
import trimesh
9-
from tqdm import tqdm
10-
11-
import b3d
12-
import b3d.chisight.dense.differentiable_renderer as rendering
1311
from b3d import Pose
12+
from tqdm import tqdm
1413

1514
rr.init("gradients")
1615
rr.connect("127.0.0.1:8812")

‎demos/differentiable_renderer/patch_tracking/demo_utils.py‎

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,10 @@
11
import os
22

3+
import b3d
4+
import b3d.utils as utils
35
import jax
46
import jax.numpy as jnp
57
import trimesh
6-
7-
import b3d
8-
import b3d.utils as utils
98
from b3d import Pose
109

1110
### Utils ###

‎demos/differentiable_renderer/patch_tracking/model.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1+
import b3d
2+
import b3d.chisight.dense.differentiable_renderer as rendering
13
import genjax
24
import jax
35
import jax.numpy as jnp
46
import rerun as rr
7+
from b3d.modeling_utils import uniform_pose
58

6-
import b3d
7-
import b3d.chisight.dense.differentiable_renderer as rendering
89
import demos.differentiable_renderer.patch_tracking.demo_utils as utils
9-
from b3d.modeling_utils import uniform_pose
1010

1111

1212
def normalize(v):

‎demos/differentiable_renderer/patch_tracking/multiple_patch_tracker_2.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
1+
import b3d
2+
import b3d.chisight.dense.patch_tracking as tracking
13
import numpy as np
24
import rerun as rr
5+
from b3d.chisight.dense.model import rr_log_uniformpose_meshes_to_image_model_trace
36
from tqdm import tqdm
47

5-
import b3d
6-
import b3d.chisight.dense.patch_tracking as tracking
78
import demos.differentiable_renderer.patch_tracking.demo_utils as du
8-
from b3d.chisight.dense.model import rr_log_uniformpose_meshes_to_image_model_trace
99

1010
rr.init("multiple_patch_tracking_2")
1111
rr.connect("127.0.0.1:8812")

‎demos/differentiable_renderer/patch_tracking/multiple_patch_tracking.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,21 +2,21 @@
22

33
import os
44

5+
import b3d
6+
import b3d.chisight.dense.differentiable_renderer as r
7+
import b3d.chisight.dense.likelihoods as l
58
import genjax
69
import jax
710
import jax.numpy as jnp
811
import numpy as np
912
import optax
1013
import rerun as rr
1114
import trimesh
15+
from b3d import Pose
1216
from tqdm import tqdm
1317

14-
import b3d
15-
import b3d.chisight.dense.differentiable_renderer as r
16-
import b3d.chisight.dense.likelihoods as l
1718
import demos.differentiable_renderer.patch_tracking.demo_utils as du
1819
import demos.differentiable_renderer.patch_tracking.model as m
19-
from b3d import Pose
2020

2121
rr.init("multiple_patch_tracking")
2222
rr.connect("127.0.0.1:8812")

‎demos/differentiable_renderer/patch_tracking/single_patch_tracking_adam.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,17 @@
11
### Preliminaries ###
22

3+
import b3d.chisight.dense.differentiable_renderer as r
4+
import b3d.chisight.dense.likelihoods as l
35
import genjax
46
import jax
57
import jax.numpy as jnp
68
import optax
79
import rerun as rr
10+
from b3d import Pose
811
from tqdm import tqdm
912

10-
import b3d.chisight.dense.differentiable_renderer as r
11-
import b3d.chisight.dense.likelihoods as l
1213
import demos.differentiable_renderer.patch_tracking.demo_utils as du
1314
import demos.differentiable_renderer.patch_tracking.model as m
14-
from b3d import Pose
1515

1616
rr.init("single_patch_tracking")
1717
rr.connect("127.0.0.1:8812")

‎demos/differentiable_renderer/patch_tracking/single_patch_tracking_mh.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,18 +2,18 @@
22

33
import time
44

5+
import b3d
6+
import b3d.chisight.dense.differentiable_renderer as r
7+
import b3d.chisight.dense.likelihoods as l
58
import genjax
69
import jax
710
import jax.numpy as jnp
811
import rerun as rr
12+
from b3d import Pose
913
from tqdm import tqdm
1014

11-
import b3d
12-
import b3d.chisight.dense.differentiable_renderer as r
13-
import b3d.chisight.dense.likelihoods as l
1415
import demos.differentiable_renderer.patch_tracking.demo_utils as du
1516
import demos.differentiable_renderer.patch_tracking.model as m
16-
from b3d import Pose
1717

1818
rr.init("single_patch_tracking-mh")
1919
rr.connect("127.0.0.1:8812")

0 commit comments

Comments
 (0)