Spaces:
Runtime error
Runtime error
add sod output
Browse files- 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, (
|
| 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')
|