wenpeng commited on
Commit
f36f09a
·
1 Parent(s): 6483e22

add sod output

Browse files
Files changed (1) hide show
  1. sod/infer_model.py +1 -1
sod/infer_model.py CHANGED
@@ -76,7 +76,7 @@ class IVModel():
76
  img_t = self.input_preprocess_tensor(img)
77
  shape = [torch.as_tensor([img_t.shape[2]]), torch.as_tensor([img_t.shape[3]])]
78
  h, w = img_t.shape[2], img_t.shape[3]
79
- img_t_temp = F.interpolate(img_t, (1024, 1024), mode='area')
80
  with torch.no_grad():
81
  res = self.net(img_t_temp, shape=shape)
82
  res = F.interpolate(res[0],size=shape, mode='bilinear')
 
76
  img_t = self.input_preprocess_tensor(img)
77
  shape = [torch.as_tensor([img_t.shape[2]]), torch.as_tensor([img_t.shape[3]])]
78
  h, w = img_t.shape[2], img_t.shape[3]
79
+ img_t_temp = F.interpolate(img_t, (512, 512), mode='area')
80
  with torch.no_grad():
81
  res = self.net(img_t_temp, shape=shape)
82
  res = F.interpolate(res[0],size=shape, mode='bilinear')