caffe 图片均值处理,网络结构图绘制,保存训练日志

一、图片均值处理

caffe使用的mean处理是为了图像像素值能更接近(0,0)原点。在caffe中自带了处理均值的文件。

sudo /opt/caffe/build/tools/comput_image_mean /opt/caffe/examples/mnist/mnist_train_lmdb /opt/caffe/mean.binaryproto

可以看到需要两个参数,一个是数据,一个是输出的文件存储位置。

二、绘制网络结构图

caffe的工具中也又绘制网络结构图。绘制结构图前需要准备环境:
sudo apt-get install graphviz

pip install pydot

然后开始绘制:
sudo python /opt/caffe/python/draw_net.py /opt/caffe/examples/mnist/lenet_train_test.solverproto /opt/caffe/lenet.png --rankdir=BT

=号后面有四种选择,BT,TB,LR,RL。

然后可以看到在第二个参数所设置的路径下有了该图片,由于我是在服务器上的docker里面跑的要把图片搬到自己的电脑上打开还要几步(心累!!!):

exit      退出当前容器,

docker ps 看看容器还在不,没在还要dockers start 容器号 来启动。

docker cp 容器号:/opt/caffe/lenet.png /home/xyj/yyc/           从容器中下载文件到服务器上,上传也是用的这个命令。

然后再用sz lenet.png命令来保存到本地上。

三、保存训练日志

可以看到例如train_lenet.sh等脚本中用到的也是build/tools中的可执行文件。

./build/tools/caffe train -solver examples/cifar10/cifar10_quick_solver.prototxt 2>&1|tee train.log

只是在最后通过重定向,管道和tee命令来保存训练日志。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值