深度学习框架caffe及py-fast-rcnn配置安装(基于CPU环境)

本文档介绍了在Ubuntu14.04系统上,如何将已配置的GPU版Py-Faster-RCNN和Caffe转换为CPU环境运行。主要涉及修改`config.py`、`test_net.py`、`train_net.py`、`setup.py`和`nms_wrapper.py`文件,注释掉GPU相关部分。完成修改后,可以成功运行demo并保存检测结果图片。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

系统环境

ubuntu14.04

说明

本来我配置的是GPU环境下的框架的,参考的是我师兄的博客,链接为:https://blog.youkuaiyun.com/PrincePaul3/article/details/52094886点击打开链接

但是在配置完成后,在运行demo时,有时候能出结果,而大部分情况下都会报错,错误如下:

Check failed: error == cudaSuccess (2 vs. 0)  out of memory
*** Check failure stack trace: ***
已放弃 (核心已转储)

显然这是我的GPU显存不够,因此我只能转为在CPU环境下运行我的demo。因为我之前配置gpu环境时,前期的编译caffe,以及编译faster-cnn已经完成,具体参考上面的链接。下面我只需要修改py-faster-rcnn相关操作,使其可以在CPU环境下运行即可。

修改步骤:

(1)将py-faster-rcnn/lib/fast_rcnn/config.py的__C.USE_GPU_NMS =

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值