会员
周边
新闻
博问
闪存
众包
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
yancx
博客园
首页
新随笔
联系
订阅
管理
随笔分类 -
深度学习-代码实现
NMS(解决大多数手写NMS运算效率较低的问题)
摘要:现在网上大多数自己手写的NMS都会导致GPU利用率较低,进而导致训练的时间较长,这是因为会经常调用单个box和其他box之间的IOU计算,导致内存和显存之间的IO交互较多。 def nms(boxes, scores, iou_threshold): keep = [] idxs = scores.
阅读全文
posted @
2022-07-31 17:47
Ayu000
阅读(434)
评论(0)
推荐(0)
公告