Hugging Face's logo

Spaces:
adirathor07
/
snap2scene
Running

App Files Community
1
adirathor07 commited on
Commit
bdfffc9
·
1 Parent(s): 5c52e2e

added test images

Browse files
Files changed (13) hide show
  1. helpers.py +4 -2
  2. sample_images_pix2vox/test_0/00.png +0 -0
  3. sample_images_pix2vox/test_0/01.png +0 -0
  4. sample_images_pix2vox/test_0/02.png +0 -0
  5. sample_images_pix2vox/test_1/00.png +0 -0
  6. sample_images_pix2vox/test_1/05.png +0 -0
  7. sample_images_pix2vox/test_1/14.png +0 -0
  8. sample_images_pix2vox/test_2/00.png +0 -0
  9. sample_images_pix2vox/test_2/20.png +0 -0
  10. sample_images_pix2vox/test_2/21.png +0 -0
  11. sample_images_pix2vox/test_2/23.png +0 -0
  12. sample_images_pix2vox/test_2/model.binvox +0 -0
  13. sample_images_pix2vox/test_2/pred.binvox +0 -0
helpers.py CHANGED
@@ -65,6 +65,10 @@ def predict_voxel_from_images(rendering_images):
65
  merger = Merger(cfg)
66
 
67
 
 
 
 
 
68
  if torch.cuda.is_available():
69
  encoder = torch.nn.DataParallel(encoder).cuda()
70
  decoder = torch.nn.DataParallel(decoder).cuda()
@@ -84,8 +88,6 @@ def predict_voxel_from_images(rendering_images):
84
 
85
 
86
 
87
- print('[INFO] %s Loading weights from %s ...' % (dt.now(), cfg.CONST.WEIGHTS))
88
- checkpoint = torch.load(cfg.CONST.WEIGHTS, weights_only=False, map_location=device)
89
 
90
  epoch_idx = checkpoint['epoch_idx']
91
  encoder.load_state_dict(encoder_state_dict)
 
65
  merger = Merger(cfg)
66
 
67
 
68
+ print('[INFO] %s Loading weights from %s ...' % (dt.now(), cfg.CONST.WEIGHTS))
69
+ checkpoint = torch.load(cfg.CONST.WEIGHTS, weights_only=False, map_location=device)
70
+
71
+
72
  if torch.cuda.is_available():
73
  encoder = torch.nn.DataParallel(encoder).cuda()
74
  decoder = torch.nn.DataParallel(decoder).cuda()
 
88
 
89
 
90
 
 
 
91
 
92
  epoch_idx = checkpoint['epoch_idx']
93
  encoder.load_state_dict(encoder_state_dict)
sample_images_pix2vox/test_0/00.png ADDED
sample_images_pix2vox/test_0/01.png ADDED
sample_images_pix2vox/test_0/02.png ADDED
sample_images_pix2vox/test_1/00.png ADDED
sample_images_pix2vox/test_1/05.png ADDED
sample_images_pix2vox/test_1/14.png ADDED
sample_images_pix2vox/test_2/00.png ADDED
sample_images_pix2vox/test_2/20.png ADDED
sample_images_pix2vox/test_2/21.png ADDED
sample_images_pix2vox/test_2/23.png ADDED
sample_images_pix2vox/test_2/model.binvox ADDED
Binary file (1.87 kB). View file
 
sample_images_pix2vox/test_2/pred.binvox ADDED
Binary file (1.82 kB). View file