通过CNN训练自己的数据集(总)

本文详细介绍了深度学习图片分类的全过程,从数据预处理到模型构建、训练及测试,探讨了如何提高分类识别率,适合对深度学习感兴趣的学习者。

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

第一步,将输入的图片分类放在InputImage中,输出的图片统一放在inputdata文件夹,并且生成tfrecord文件。
第二步,定义了几个函数,定义的函数在第五步测试时候被调用
第三步,定义模型,没有输入,也没有输出,只是定义了几个函数

第四步,训练,生成ckpt文件,确定两个路径,分别是

train_dir 即shuffle后放在同一个文件夹的路径
logs_train_dir 即CKPT文件保存路径

    为什么val_writer被注释掉?    

第五步,测试,输入两个路径,train_dir 即shuffle后放在同一个文件夹的路径 logs_train_dir 即CKPT文件保存路径

输出为图片及其准确率。

 

test函数需要改进,改为批量读取。同时弄清楚test的原理,改变文件夹方式
              对训练用的图片生成测试集或者验证集,那训练时候怎么对输入的图片进行处理的呢?
    get_one_image图片需要处理下

并非使用模型训练,而是自己构建模型,训练出参数,然后保存参数,进行测试

来自之前调试的机车训练模型半成品


主要问题:分类识别率在50%,有点低。这个概率是否正确


改进:也许不正确,因为是根据softmax来的。只要大于0.5就说明准确率高呢。可以手工验证,多次读取图片,机器进行判断,然后查看
1、

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值