ZhengPeng7 commited on
Commit
264c03c
·
verified ·
1 Parent(s): d5ee825

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,6 +69,7 @@ def pred_maps(images):
69
  scaled_preds_tensor = model(images_proc.to(device))[-1]
70
  preds = []
71
  print('scaled_preds_tensor.sum():', scaled_preds_tensor.sum())
 
72
  for image_shape, pred_tensor, save_path in zip(image_shapes, scaled_preds_tensor, save_paths):
73
  if device == 'cuda':
74
  pred_tensor = pred_tensor.cpu()
 
69
  scaled_preds_tensor = model(images_proc.to(device))[-1]
70
  preds = []
71
  print('scaled_preds_tensor.sum():', scaled_preds_tensor.sum())
72
+ print('type(scaled_preds_tensor):', type(scaled_preds_tensor))
73
  for image_shape, pred_tensor, save_path in zip(image_shapes, scaled_preds_tensor, save_paths):
74
  if device == 'cuda':
75
  pred_tensor = pred_tensor.cpu()