Ubuntu+caffe训练cifar-10数据集

本文介绍如何使用CIFAR-10数据集进行图像分类任务的训练流程,包括数据集下载、LMDB及均值文件生成、配置训练模式(CPU/GPU)、执行训练等步骤。

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

1. 下载cifar-10数据库


ciffar-10数据集包含10种物体分类,50000张训练图片,10000张测试图片。

在终端执行指令下载cifar-10数据集(二进制文件):


[html]  view plain  copy
 print ?
  1. cd ~/caffe-master  
  2. ./data/cifar10/get_cifar10.sh  


在./data/cifar10文件夹下生成5个.bin的训练数据集合1个测试数据集:




2. 生成lmdb以及均值文件


[html]  view plain  copy
 print ?
  1. ./examples/cifar10/create_cifar10.sh  


执行之后在./examples/cifar10下生成训练和测试lmdb文件和均值mean.binaryproto文件


3. 根据系统修改运行模式(CPU或GPU)


[html]  view plain  copy
 print ?
  1. sudo gedit ./examples/cifar10/cifar10_quick_solver.prototxt  

修改最后一行 solver_mode 的配置为CPU或GPU


4. 执行训练


执行cifar10文件夹下的train_quick.sh开始训练:


[html]  view plain  copy
 print ?
  1. ./examples/cifar10/train_quick.sh  




4000次的迭代在CPU上训练耗时约30min,最终精度是84%:


转载于:https://www.cnblogs.com/mtcnn/p/9411828.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值