本文整理匯總了Python中nms.gpu_nms.gpu_nms方法的典型用法代碼示例。如果您正苦於以下問題：Python gpu_nms.gpu_nms方法的具體用法？Python gpu_nms.gpu_nms怎麽用？Python gpu_nms.gpu_nms使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類nms.gpu_nms的用法示例。
在下文中一共展示了gpu_nms.gpu_nms方法的7個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: nms
# 需要導入模塊: from nms import gpu_nms [as 別名]
# 或者: from nms.gpu_nms import gpu_nms [as 別名]
def nms(dets, thresh, force_cpu=False):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
if cfg.USE_GPU_NMS and not force_cpu:
return gpu_nms(dets, thresh, device_id=0)
else:
return cpu_nms(dets, thresh) 示例2: nms
# 需要導入模塊: from nms import gpu_nms [as 別名]
# 或者: from nms.gpu_nms import gpu_nms [as 別名]
def nms(dets, thresh, force_cpu=False):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
if cfg.USE_GPU_NMS and not force_cpu:
from nms.gpu_nms import gpu_nms
return gpu_nms(dets, thresh, device_id=0)
else:
from nms.cpu_nms import cpu_nms
return cpu_nms(dets, thresh) 示例3: nms
# 需要導入模塊: from nms import gpu_nms [as 別名]
# 或者: from nms.gpu_nms import gpu_nms [as 別名]
def nms(dets, thresh, force_cpu=False):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
if cfg.USE_GPU_NMS and not force_cpu:
return gpu_nms(dets, thresh, device_id=cfg.GPU_ID)
else:
return cpu_nms(dets, thresh) 示例4: nms
# 需要導入模塊: from nms import gpu_nms [as 別名]
# 或者: from nms.gpu_nms import gpu_nms [as 別名]
def nms(dets, thresh, force_cpu=False):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
if cfg.USE_GPU_NMS and not force_cpu:
return gpu_nms(dets, thresh, device_id=cfg.GPU_ID)
else:
return cpu_nms(dets, thresh) 示例5: nms
# 需要導入模塊: from nms import gpu_nms [as 別名]
# 或者: from nms.gpu_nms import gpu_nms [as 別名]
def nms(dets, thresh, force_cpu=False):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
if cfg.USE_GPU_NMS and not force_cpu:
return gpu_nms(dets, thresh, device_id=0)
else:
return cpu_nms(dets, thresh) 示例6: apply_nms
# 需要導入模塊: from nms import gpu_nms [as 別名]
# 或者: from nms.gpu_nms import gpu_nms [as 別名]
def apply_nms(cls_dets, nms_method, nms_thresh):
# nms and filter
keep = np.where((cls_dets[:, 4] >= min_scores) &
((cls_dets[:, 3] - cls_dets[:, 1]) * (cls_dets[:, 2] - cls_dets[:, 0]) >= min_box_size))[0]
cls_dets = cls_dets[keep]
if len(cls_dets) > 0:
if nms_method == 'nms':
keep = gpu_nms(cls_dets, nms_thresh)
elif nms_method == 'soft':
keep = cpu_soft_nms(np.ascontiguousarray(cls_dets, dtype=np.float32), method=2)
else:
assert False
cls_dets = cls_dets[keep]
return cls_dets, keep 示例7: nms
# 需要導入模塊: from nms import gpu_nms [as 別名]
# 或者: from nms.gpu_nms import gpu_nms [as 別名]
def nms(dets, thresh):
"""Dispatch to either CPU or GPU NMS implementations."""
if dets.shape[0] == 0:
return []
if cfg.USE_GPU_NMS:
return gpu_nms(dets, thresh, device_id=cfg.GPU_ID)
else:
return cpu_nms(dets, thresh)