Commit ca7e0f11 authored by Baumgartner, Michael's avatar Baumgartner, Michael
Browse files

Merge branch '0000_project' of https://github.com/MIC-DKFZ/nnDetection into 0000_project

parents 73c0a226 96ed0bd7
...@@ -15,11 +15,16 @@ limitations under the License. ...@@ -15,11 +15,16 @@ limitations under the License.
""" """
import torch import torch
from loguru import logger
from torch import Tensor from torch import Tensor
from torch.cuda.amp import autocast from torch.cuda.amp import autocast
from torchvision.ops.boxes import nms as nms_2d from torchvision.ops.boxes import nms as nms_2d
from nndet._C import nms as nms_gpu try:
from nndet._C import nms as nms_gpu
except ImportError:
logger.warning("nnDetection was not build with GPU support!")
nms_gpu = None
from nndet.core.boxes.ops import box_iou from nndet.core.boxes.ops import box_iou
......
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