深度学习
楚兴
这个作者很懒,什么都没留下…
展开
-
Caffe nvcc编译问题:Unsupported gpu architecture
最近学习DSSD,从https://github.com/chengyangfu/caffe/tree/dssd下载源码后编译报错:Unsupported gpu architecture 'compute_20',具体如下: CXX src/caffe/util/db.cpp CXX src/caffe/util/benchmark.cpp CXX src/caffe/util/im2col.c...原创 2018-11-05 21:43:12 · 3227 阅读 · 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 1 In file inc...原创 2018-11-08 22:26:12 · 1022 阅读 · 0 评论 -
Caffe SSD编译、训练及测试
SSD采用VGG16作为基础模型,然后在VGG16的基础上新增了卷积层来获得更多的特征以用于检测。SSD的网络结构如上图所示(上面是SSD模型,下面是Yolo模型),可以明显看到SSD利用了多尺度的特征图做检测。 安装 clone代码(假设代码clone到$CAFFE_ROOT目录)。 git clone https://github.com/weiliu89/caffe.git cd ca...原创 2018-11-25 23:10:07 · 2918 阅读 · 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 · 1167 阅读 · 0 评论