【YOLO】YOLO: You Only Look Once GPU版本的配置与实现

博主对比了TensorFlow和darknet实现的YOLO在GPU上的性能,发现darknet在GPU工作站上运行速度显著提高。通过详细阐述如何修改MakeFile配置和安装nvidia-cuda-toolkit,实现了GPU加速,最终达到60.496 FPS的帧率,适合实时目标检测应用。
摘要由CSDN通过智能技术生成

Update: 博主建了一个计算机视觉&YOLO的群:659389932。希望与大家多多交流。(2017-08-15)

说明

使用TensorFlow实现的YOLO,发现效果很一般(GTX960下,一张图1.2s左右),因此打算用darknet实现的YOLO来跑跑看效果。


大致交代一下,TensorFlow版本的YOLO在台式机显卡GTX960,16GB内存情况下,大概是1.2s一张图。现在在GPU工作站(CPU至强系列,GPU两块Titan xp)上测试用darknet实现的YOLO,基于CPU的是12.3s一张图,这个效果是真的很差啊,然后就想既然都用了GPU工作站了,那用GPU跑起来。



官网(

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值