lufficc
diff --git a/‎ssd/utils/nms.py‎
Lines changed: 1 addition & 4 deletions b/‎ssd/utils/nms.py‎
Lines changed: 1 addition & 4 deletions
diff --git a/‎ssd/utils/python_nms.py‎
Lines changed: 0 additions & 59 deletions b/‎ssd/utils/python_nms.py‎
Lines changed: 0 additions & 59 deletions
@@ -9,10 +9,7 @@
     if torchvision.__version__ >= '0.3.0':
         _nms = torchvision.ops.nms
     else:
-        from .python_nms import python_nms
-
-        _nms = python_nms
-        warnings.warn('You are using python version NMS, which is very very slow. Try compile c++ NMS '
+        warnings.warn('No NMS is available. Please upgrade torchvision or compile c++ NMS '
                       'using `cd ext & python build.py build_ext develop`')