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

+1-2
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

+1-2
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

+3-4
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

+3-4
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

+2-3
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

+3-3
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

+3-3
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

+4-4
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

+3-3
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

+4-4
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")

‎demos/differentiable_renderer/test_barycentric_interp.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
1-
import jax
2-
import jax.numpy as jnp
3-
import rerun as rr
4-
51
import b3d
62
import b3d.chisight.dense.differentiable_renderer as rendering
73
import b3d.chisight.dense.likelihoods as likelihoods
4+
import jax
5+
import jax.numpy as jnp
6+
import rerun as rr
87
from b3d import Pose
98

109
# Set up OpenGL renderer

‎demos/differentiable_renderer/test_basic_gd.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
import time
22

3+
import b3d
4+
import b3d.chisight.dense.differentiable_renderer as rendering
5+
import b3d.chisight.dense.likelihoods as likelihoods
36
import genjax
47
import jax
58
import jax.numpy as jnp
69
import rerun as rr
710

8-
import b3d
9-
import b3d.chisight.dense.differentiable_renderer as rendering
10-
import b3d.chisight.dense.likelihoods as likelihoods
1111
import demos.differentiable_renderer.utils as utils
1212

1313
# Set up OpenGL renderer

‎demos/differentiable_renderer/utils.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
1+
import b3d
12
import jax
23
import jax.numpy as jnp
3-
4-
import b3d
54
from b3d import Pose
65

76

‎demos/fork_knife_smc_identity_pose.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,14 @@
11
import os
22

3+
import b3d
34
import genjax
45
import jax
56
import jax.numpy as jnp
67
import numpy as np
78
import rerun as rr
89
import trimesh
9-
from tqdm import tqdm
10-
11-
import b3d
1210
from b3d import Pose
11+
from tqdm import tqdm
1312

1413
### Choose experiment
1514

‎demos/graphics_edits_demo/demo_visualize.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,14 @@
22
import os
33
import pickle
44

5+
import b3d
56
import jax
67
import jax.numpy as jnp
78
import numpy as np
89
import rerun as rr
910
import trimesh
10-
from tqdm import tqdm
11-
12-
import b3d
1311
from b3d import Pose
12+
from tqdm import tqdm
1413

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

‎demos/graphics_edits_demo/vkm_demo.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
11
import os
22

3+
import b3d
34
import jax
45
import jax.numpy as jnp
56
import numpy as np
67
import rerun as rr
78
import trimesh
8-
9-
import b3d
109
from b3d import Pose
1110

1211
rr.init("vkm_demo2")

‎demos/mesh_fitting/demo.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
import os
22

3+
import b3d
34
import genjax
45
import jax
56
import jax.numpy as jnp
67
import optax
78
import rerun as rr
89
from tqdm import tqdm
910

10-
import b3d
1111
import demos.mesh_fitting.model as m
1212
import demos.mesh_fitting.tessellation as t
1313

‎demos/mesh_fitting/demo_depth_init.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
import os
22

3+
import b3d
34
import genjax
45
import jax
56
import rerun as rr
67

7-
import b3d
88
import demos.mesh_fitting.model as m
99
import demos.mesh_fitting.utils as u
1010

‎demos/mesh_fitting/model.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
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
5-
6-
import b3d
7-
import b3d.chisight.dense.differentiable_renderer as rendering
87
from b3d.modeling_utils import uniform_pose
98

109

‎demos/mesh_fitting/utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1+
import b3d
12
import jax
23
import jax.numpy as jnp
34

4-
import b3d
55
import demos.mesh_fitting.tessellation as t
66

77

‎demos/posterior_datasets/identity_posterior_data_gen.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,12 @@
11
import os
22

3+
import b3d
34
import jax
45
import jax.numpy as jnp
56
import numpy as np
67
import rerun as rr
78
import trimesh
89

9-
import b3d
10-
1110
width = 200
1211
height = 200
1312
fx = 300.0

‎demos/posterior_datasets/pose_posterior_data_gen.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,10 @@
11
import os
22

3+
import b3d
34
import jax
45
import jax.numpy as jnp
56
import rerun as rr
67
import trimesh
7-
8-
import b3d
98
from b3d import Pose
109

1110
rr.init("demo")

‎demos/sparse_model/cotracker.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,10 @@
11
import argparse
22
import time
33

4+
import b3d
45
import numpy as np
56
import torch
67

7-
import b3d
8-
98
parser = argparse.ArgumentParser("r3d_to_video_input")
109
parser.add_argument("input", help=".r3d File", type=str)
1110
args = parser.parse_args()

‎demos/sparse_model/sparse_model.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,14 @@
11
import os
22
from functools import partial
33

4+
import b3d
45
import jax
56
import jax.numpy as jnp
67
import optax
78
import rerun as rr
89
import trimesh
910
from tqdm import tqdm
1011

11-
import b3d
12-
1312

1413
def map_nested_fn(fn):
1514
"""Recursively apply `fn` to the key-value pairs of a nested dict."""

‎demos/sparse_model/sparse_model_cotracker.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import os
22
from functools import partial
33

4+
import b3d
45
import jax
56
import jax.numpy as jnp
67
import numpy as np
@@ -9,8 +10,6 @@
910
from matplotlib import colormaps
1011
from tqdm import tqdm
1112

12-
import b3d
13-
1413

1514
def map_nested_fn(fn):
1615
"""Recursively apply `fn` to the key-value pairs of a nested dict."""

0 commit comments

Comments
 (0)