![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
楚兴
这个作者很懒,什么都没留下…
展开
-
Caffe nvcc编译问题:Unsupported gpu architecture
最近学习DSSD,从https://github.com/chengyangfu/caffe/tree/dssd下载源码后编译报错:Unsupported gpu architecture 'compute_20',具体如下:CXX src/caffe/util/db.cppCXX src/caffe/util/benchmark.cppCXX src/caffe/util/im2col.c...原创 2018-11-05 21:43:12 · 3195 阅读 · 0 评论 -
Caffe 运行平台支持DSSD模型
DSSD源码在https://github.com/chengyangfu/caffe/tree/dssd,在cuda9环境下编译,首先把Makefile.config里面的USE_CUDNN := 1开关打开,编译报如下的错:make: *** [.build_release/src/caffe/layers/multibox_loss_layer.o] Error 1In file inc...原创 2018-11-08 22:26:12 · 997 阅读 · 0 评论 -
Caffe SSD编译、训练及测试
SSD采用VGG16作为基础模型,然后在VGG16的基础上新增了卷积层来获得更多的特征以用于检测。SSD的网络结构如上图所示(上面是SSD模型,下面是Yolo模型),可以明显看到SSD利用了多尺度的特征图做检测。安装clone代码(假设代码clone到$CAFFE_ROOT目录)。git clone https://github.com/weiliu89/caffe.gitcd ca...原创 2018-11-25 23:10:07 · 2881 阅读 · 0 评论 -
Caffe 运行平台支持DenseNet模型
文章目录1 集成DenseNet的报错信息2 caffe源码分析及修改2.1 pooling层源码分析2.2 caffe源码修改3 总结在计算机视觉领域,卷积神经网络(CNN)已经成为最主流的方法,比如GoogLenet、VGG-19、Incepetion等模型。CNN史上的一个里程碑事件是ResNet模型的出现,ResNet可以训练出更深的CNN模型,从而实现更高的准确度。ResNet模型的核...原创 2019-03-23 23:39:22 · 1111 阅读 · 0 评论