
问题总结
htshinichi
社畜
展开
-
【caffe】配置caffe记录(GPU)[2018.11.07更新]
配置过程必须记录下来,否则换台电脑继续踩坑懵逼。 文章目录电脑配置安装CUDA 电脑配置 系统:Ubuntu16.04 GPU:NVIDIA GeForce 940MX Python版本:3.5.2(系统自带,未使用anaconda) 可以通过lspci | grep -i nvidia来查看显卡型号。由于我的显卡驱动是装好的,因此跳过。有需要可以参考网上安装驱动教程。通过sudo dpkg ...原创 2018-10-15 11:45:08 · 1510 阅读 · 0 评论 -
【caffe】根据prototxt修改caffemodel
需求:想要在已训练或预训练的caffe模型中将网络结构prototxt删掉几层,并且将这几层的参数从caffemodel中也删除(这样可以减小caffemodel的大小)。 import caffe import caffe_pb2 BEFORE_MODIFY_DEPLOY_NET = "yourpath/before_modify_deploy.prototxt" AFTER_MODIFY_DE...原创 2018-12-18 10:46:11 · 3347 阅读 · 0 评论