(一) cifar10分类训练

本文档介绍了如何使用Caffe框架在Ubuntu系统上处理CIFAR-10数据库,包括下载、解压数据,将二进制文件转换为lmdb格式,创建训练所需的数据库均值文件,最后详细阐述了训练过程和测试步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

       本次学习使用caffe-master文件在Downloads目录下,使用caffe-master

1.准备数据库  CIFAR-10(默认在caffe-master目录下)

终端执行如下命令获取数据库文件

chenshaobo@idea:~/Downloads/caffe-master$ cd data/cifar10/

chenshaobo@idea:~/Downloads/caffe-master/data/cifar10$ ./get_cifar10.sh

执行之后,当前目录下有以下文件:(ls)

        batches.meta.txt  data_batch_3.bin  get_cifar10.sh
       data_batch_1.bin  data_batch_4.bin  readme.html
      data_batch_2.bin  data_batch_5.bin  test_batch.bin

其中shell文件解析如下:

DIR="$( cd "$(dirname "$0")" ; pwd -P )"     //dirname "$0"是获取脚本所在的目录;pwd -P可以获取当前目录的绝对路径;
cd $DIR
 
echo "Downloading..."
 
wget --no-check-certificate http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz
 
echo "Unzipping..."
 
tar -xf cifar-10-binary.tar.gz && rm -f cifar-10-binary.tar.gz
mv cifar-10-batches-bin/* . && rm -rf cifar-10-batches-bin

表示获取文件解压之后的cifar-10-batches-bin文件夹下的所有文件都移到当前目录下并删除空文件夹;

然后,执行一下命令将获取的二进制文件转化为lmdb格式文件并计算image mean,如下:

chenshaobo@idea:~/Downloads/caffe-master/data/cifar10$ cd ../..
chenshaobo@idea:~/Downloads/caffe-master$ ./examples/cifar10/create_cifar10.sh 

运行完之后,将会在examples/cifar10下生成两个lmdb格式文件和数据库图像均值二进制文件。

2.训练和测试

训练主要使用网络配置文件cifar10_quick_train_test.prototxt和网络求解文件cifar10_quick_solver.prototxt

chenshaobo@idea:~/Downloads/caffe-master/examples/cifar10$ cd ../..
chenshaobo@idea:~/Downloads/caffe-master$ ./examples/cifar10/train_quick.sh

执行完命令后,此脚本首先进行网络的搭建工作。

然后,才开始进行训练。

通过设置后,每100次迭代会显示一次训练时的学习率lr和损失函数loss,每500次测试一次,输出#0(准确率)和*1(测试损失函数)。当迭代完之后,可以获得此模型的准确率。其中,模型的参数以二进制protobuf格式形式进行存储。


然后利用此模型训练新数据。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值