CUDA
Law-Yao
数字信号处理与人工智能相关的软硬件系统实现工作。
展开
-
CUDA版本的Locality-aware NMS
EAST文本检测涉及的Locality-aware NMS,通过合并同类型的倾斜框(IOU高于一定的阈值),能够提高长文本框检测的有效性,但CPU版本的Locality-aware NMS执行速度较慢。CUDA版本的Locality-aware NMS如下:#include "rotate_gpu_nms.hpp"#include <vector>#include <i...原创 2019-11-30 13:15:29 · 1076 阅读 · 3 评论 -
转载:简明实用的CUDA入门教程
“CUDA Tutorial”Sample code in adding 2 numbers with a GPUTerminology: Host (a CPU and host memory), device (a GPU and device memory).This sample code adds 2 numbers together with a GPU:Define a kernel...转载 2018-05-25 16:22:14 · 1500 阅读 · 1 评论 -
MXNet的高效率CUDA NMS解析
非极大抑制(NMS:Non Maximum Suppression)起到边框(水平框或倾斜框)去重叠的作用,广泛应用于通用目标检测、人脸检测与OCR检测等算法的后处理。简单的Python实现如下:def standard_nms(S, thres): order = np.argsort(S[:, -1])[::-1] keep = [] while order.si...原创 2019-06-13 11:51:41 · 1796 阅读 · 0 评论