gherget commited on
Commit
44f25e9
1 Parent(s): 40250f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -86,9 +86,8 @@ def predict(net, inputs_val, shapes_val, hypar, device):
86
 
87
  ds_val = net(inputs_val_v)[0] # list of 6 results
88
 
89
- print(ds_val)
90
  print(ds_val[0])
91
- pred_val = ds_val[1][0,:,:,:] # B x 1 x H x W # we want the first one which is the most accurate prediction
92
 
93
  ## recover the prediction spatial size to the orignal image size
94
  pred_val = torch.squeeze(F.upsample(torch.unsqueeze(pred_val,0),(shapes_val[0][0],shapes_val[0][1]),mode='bilinear'))
 
86
 
87
  ds_val = net(inputs_val_v)[0] # list of 6 results
88
 
 
89
  print(ds_val[0])
90
+ pred_val = ds_val[5][0,:,:,:] # B x 1 x H x W # we want the first one which is the most accurate prediction
91
 
92
  ## recover the prediction spatial size to the orignal image size
93
  pred_val = torch.squeeze(F.upsample(torch.unsqueeze(pred_val,0),(shapes_val[0][0],shapes_val[0][1]),mode='bilinear'))