caffe-Faster-rcnn
Blackrosetian
热爱编程,喜欢捣腾新事物。
展开
-
安装Caffe与Faster RCNN遇到的问题汇总解决方案(GPU版)
历时一天的时间终于把Caffe和Faster RCNN搞定了,关于安装Caffe与Faster RCNN之前的cuda与cudnn请自行百度安装,安装的版本必须匹配起来,否则后面会出现各种意想不到的问题。 安装Caffe遇到的问题如下: 1、Make报错显示:nvcc fatal : Unsupported gpu architecture 'compute_20' 此问题需要修改Makef...原创 2018-12-21 11:42:53 · 430 阅读 · 0 评论 -
使用Faster RCNN训练自己的数据
数据准备 先将自己的数据转成VOC数据集类似的数据,图片为jpg格式,标签为xml格式。 在Faster RCNN目录下的data文件夹下建立VOCdevkit2007文件夹,再在此文件夹下建立VOC2007文件夹,再在VOC2007文件夹下建立Annotations,ImageSets , JPEGImages三个文件夹,Annotations文件夹下存放xml文件,ImageSets文件夹...原创 2018-12-24 09:19:26 · 1423 阅读 · 0 评论 -
配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决
解决方法:将ssd_pascal.py文件中gpus = "0,1,2,3"的GPU选择改为gpus = "0",后面的1,2,3都删掉即可。因为电脑只安装了一块GPU。还需要将ssd_pascal.py文件中中的batch_size = 32和accum_batch_size = 32同时改为8,不然会出现“Check failed: error == cudaSuccess (2 vs. 0)...原创 2019-01-10 17:01:01 · 776 阅读 · 0 评论