jev-aleks commited on
Commit
f61bfbf
·
1 Parent(s): ff4c202

remove dataset loading in demo

Browse files
Files changed (2) hide show
  1. app.py +2 -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, test_dataset = 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
 
 
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(