Skip to content

M
maskrcnn
Commit b3d1de00 authored by Csaba Botos's avatar Csaba Botos Committed by Francisco Massa
Browse files
Options

use internal interpolate implementation (#559)

parent 558d7afa
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py
View file @ b3d1de00
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
import numpy as np import numpy as np
import torch import torch
from torch import nn from torch import nn
import torch.nn.functional as F from maskrcnn_benchmark.layers.misc import interpolate
from maskrcnn_benchmark.structures.bounding_box import BoxList from maskrcnn_benchmark.structures.bounding_box import BoxList
...@@ -132,7 +132,7 @@ def paste_mask_in_image(mask, box, im_h, im_w, thresh=0.5, padding=1): ...@@ -132,7 +132,7 @@ def paste_mask_in_image(mask, box, im_h, im_w, thresh=0.5, padding=1):
# Resize mask # Resize mask
mask = mask.to(torch.float32) mask = mask.to(torch.float32)
mask = F.interpolate(mask, size=(h, w), mode='bilinear', align_corners=False) mask = interpolate(mask, size=(h, w), mode='bilinear', align_corners=False)
mask = mask[0][0] mask = mask[0][0]
if thresh >= 0: if thresh >= 0:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment