darknet推理-基于cupy

我要开发同款
db_png_jpg2023年09月12日
69阅读
所属分类人工智能IT

作品详情

使用cupy实现卷积、池化、双线性插值上采样,能够在gpu上推理yolov3、yolov4。(基于darknet格式的cfg和weights)卷积、池化均用imcol + gemm实现。推理速度和pytorch相当。
查看全文
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论