mirror of https://github.com/YifanXu74/MQ-Det.git
15 lines
325 B
Python
15 lines
325 B
Python
|
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
|
||
|
|
from maskrcnn_benchmark import _C
|
||
|
|
|
||
|
|
try:
|
||
|
|
import torchvision
|
||
|
|
from torchvision.ops import nms
|
||
|
|
except:
|
||
|
|
nms = _C.nms
|
||
|
|
|
||
|
|
ml_nms = _C.ml_nms
|
||
|
|
soft_nms = _C.soft_nms
|
||
|
|
|
||
|
|
# nms.__doc__ = """
|
||
|
|
# This function performs Non-maximum suppresion"""
|