def nms(dets, thresh, force_cpu=False):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
# ---numpy version---
# original: return gpu_nms(dets, thresh, device_id=cfg.GPU_ID)
# ---pytorch version---
return nms_gpu(dets, thresh)
def nms(dets, thresh, force=False):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
# ---numpy version---
if force == False:
return nms_cpu(dets, thresh)
else:
# ---pytorch version---
return nms_gpu(dets, thresh)
def nms(dets, thresh, force_cpu=False):
'''
:param dets: shape=(2000,5) 5=[x1,y1,x2,y2,score]
:param thresh:
:param force_cpu:
:return:
'''
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
# ---numpy version---
# original: return gpu_nms(dets, thresh, device_id=cfg.GPU_ID)
# ---pytorch version---
return nms_gpu(dets, thresh) if force_cpu == False else nms_cpu(
dets, thresh)
