Spaces:
Running
on
Zero
Running
on
Zero
remove dataset loading in demo
Browse files- app.py +2 -2
- demo_utils/utils.py +2 -2
app.py
CHANGED
@@ -29,12 +29,12 @@ download_scenedino_checkpoint("ssc-kitti-360-dinov2")
|
|
29 |
# Load model, ray sampler, datasets
|
30 |
ckpt_path = "out/scenedino-pretrained/seg-best-dino/"
|
31 |
ckpt_name = "checkpoint.pt"
|
32 |
-
net_v1, renderer_v1, ray_sampler_v1
|
33 |
renderer_v1.eval()
|
34 |
|
35 |
ckpt_path = "out/scenedino-pretrained/seg-best-dinov2/"
|
36 |
ckpt_name = "checkpoint.pt"
|
37 |
-
net_v2, renderer_v2, ray_sampler_v2
|
38 |
renderer_v2.eval()
|
39 |
|
40 |
|
|
|
29 |
# Load model, ray sampler, datasets
|
30 |
ckpt_path = "out/scenedino-pretrained/seg-best-dino/"
|
31 |
ckpt_name = "checkpoint.pt"
|
32 |
+
net_v1, renderer_v1, ray_sampler_v1 = load_modules(ckpt_path, ckpt_name)
|
33 |
renderer_v1.eval()
|
34 |
|
35 |
ckpt_path = "out/scenedino-pretrained/seg-best-dinov2/"
|
36 |
ckpt_name = "checkpoint.pt"
|
37 |
+
net_v2, renderer_v2, ray_sampler_v2 = load_modules(ckpt_path, ckpt_name)
|
38 |
renderer_v2.eval()
|
39 |
|
40 |
|
demo_utils/utils.py
CHANGED
@@ -55,9 +55,9 @@ def load_modules(
|
|
55 |
model.load_state_dict(cp, strict=False)
|
56 |
model = model.to(device)
|
57 |
|
58 |
-
test_dataset = make_datasets(config["dataset"])[1]
|
59 |
|
60 |
-
return net, renderer, ray_sampler, test_dataset
|
61 |
|
62 |
|
63 |
def load_sample_from_path(
|
|
|
55 |
model.load_state_dict(cp, strict=False)
|
56 |
model = model.to(device)
|
57 |
|
58 |
+
# test_dataset = make_datasets(config["dataset"])[1]
|
59 |
|
60 |
+
return net, renderer, ray_sampler # , test_dataset
|
61 |
|
62 |
|
63 |
def load_sample_from_path(
|