def nms(dets, thresh):
if dets.shape[0] == 0:
return []
try:
return gpu_nms(dets, thresh, device_id=0)
except:
return cython_nms(dets, thresh)
def nms(dets, thresh):
if dets.shape[0] == 0:
return []
try:
if GPU and GPUID is not None:
return gpu_nms(dets, thresh, device_id=GPUID)
except:
return cython_nms(dets, thresh)