A-Fast-RCNN代码运行以及问题解决方法

本文分享了A-Fast-RCNN:HardPositiveGenerationviaAdversaryforObjectDetection论文的实践过程,包括代码克隆、环境搭建、数据集准备及训练过程中遇到的问题及其解决方案。

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

最近读了A-Fast-RCNN: Hard Positive Generation via Adversary for Object Detection这篇论文,读了之后根据作者提供的代码,准备跑一跑作者的代码。但是作者给的代码只有ASDN网络部分的,ASTN网络部分的还没有给出来,先说说自己跑代码的时候遇到的一些坑以及代码的运行过程。 

首先是从官网克隆代码:

  git clone https://github.com/xiaolonw/adversarial-frcnn

得到源码之后,要注意的是还要克隆fast RCNN的源码

  git clone https://github.com/rbgirshick/fast-rcnn(运行代码需要里面的ss算法部分)

在运行作者的源码之前,要准备安装caffe,caffe安装请参照网上博客,不同的人会出现不同的问题,安装完之后记得添加到环境变量

,不然会出错。训练代码前需要下载VOC数据集,http://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar

和http://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar。下载完之后放入指定的文件中然后就可以运行训练代码了。

然后cd到代码的文件夹中,使用 sh train.sh命令来运行训练过程,整个代码只需要运行这么一句就行了。但是这个时候会提示一个

有关utils.cython_bbox模块的错误,提示没有这个模块。这时候的解决办法就是cd到/lib文件夹中,将setup.py文件中的

utils.cython_bbox替换成utils.bbox,然后make,然后依次执行Python setup.py build, Python setup.py install

如果还是出现cython_bbox的错误,直接将出错误的文件中的cython_bbox改成bbox就行。

最后再执行sh train.sh文件进行训练。目前我的代码证在跑,由于电脑内存不够出现核心已转储的错误。


评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值