计算机视觉:用inception-v3模型重新训练自己的数据模型(代码片段)

zhml zhml     2023-04-09     163

关键词:

用inception-v3重新训练自己的数据模型

背景:

现代的图像识别模型具有数以百万计的参数,从头开始训练(Train from scratch)需要大量的样本数据以及消耗巨大的计算资源(几百个GPU),因此采用迁移学习的方式重训一个模型(Retrain a model)对于学习模型的成本较低,利用Inception-V3作为已经训练好的模型,来实现自己的图像分类识别
 
 

inception模型文件目录介绍:

技术图片

data目录:需要训练的数据放在该目录:       inception_model:把下载inception模型放在该目录下:

技术图片   技术图片

 

test_images:把训练完成之后,测试图片所放目录

 

数据准备完成之后,按以下步骤执行:

 步骤1:

windows:

运行批处理文件retrain.bat

 

python retrain.py ^
--bottleneck_dir bottleneck ^      #自行生成每个训练图片的数据
--how_many_training_steps 200 ^ #步数
--model_dir inception_model ^    #模型
--output_graph output_graph.pb ^ #训练后输出模型
--output_labels output_labels.txt ^ #训练后输出label
--image_dir data
pause    

 

Ubuntu:

运行批处理文件retrain.sh

 

步骤2:

训练完成后,测试如下

 

 

Python  predict.py

测试结果如下:

技术图片

 

 

 

 

结论:如果测试物体不相似,识别率很高,但是如果物体形状相似,识别率则不高

源码:https://pan.baidu.com/s/1qdRmnQsRv5k3QesZxRC9QA    提取码:jipt

 

 

 

 

 

 

 

 

 

 

超越cnn的vit模型及其应用前景

...练模型BERT、GPT-3在NLP各个任务中都大放异彩,令人瞩目。计算机视觉分析任务也借鉴了Transformer模型的思想和方法,Meta公司的DETR模型中就使用Transformer和端到端方法实现了CV领域的目标检测任务,之后Google公司又推出了用纯粹的T... 查看详情

深度学习|计算机视觉环境配置-数据集制作-模型训练一站式教程(ubuntu系统)//

深度学习|计算机视觉环境配置-数据集制作-模型训练一站式教程(Ubuntu系统)文章目录深度学习|计算机视觉环境配置-数据集制作-模型训练一站式教程(Ubuntu系统)一.总体流程二.分步细节2.1环境配置2.1.1Pycharm持续更新一.总体流程基... 查看详情

tensorflow1.0学习:用别人训练好的模型来进行图像分类

谷歌在大型图像数据库ImageNet上训练好了一个Inception-v3模型,这个模型我们可以直接用来进来图像分类。下载地址:https://storage.googleapis.com/download.tensorflow.org/models/inception_dec_2015.zip下载完解压后,得到几个文件:其中的classify_imag... 查看详情

『计算机视觉』mask-rcnn_训练网络其三:model准备

...构的网络。如上所示,我们首先建立图结构(详见上节『计算机视觉』Mask-RCNN_训练网络其二:train网络结构),然后选择初始化参数方案例子(train_shape.ipynb)中使用的是COCO预训练模型,如果想要"Fin 查看详情

深度学习——计算机视觉

提高模型泛化能力的方法——图像增广和微调图像增广图像增广(imageaugmentation)技术通过对训练图像做一系列随机改变,来产生相似但又不同的训练样本,从而扩大训练数据集的规模。图像增广的另一种解释是&#... 查看详情

如何高效精准地进行图片搜索?看看轻量化视觉预训练模型

...发展。近来,基于Transformer的视觉预训练模型在诸多计算机视觉任务上都取得了优越性能,受到了广泛关注。然而,视觉Transformer预训练模型通常参数量大、复杂度高,制约了其在实际应用中的部署和使用,尤... 查看详情

计算机视觉(十四):tensorflow分布式训练(代码片段)

计算机视觉笔记总目录当我们拥有大量计算资源时,通过使用合适的分布式策略,我们可以充分利用这些计算资源,从而大幅压缩模型训练的时间。针对不同的使用场景,TensorFlow在tf.distribute.Strategy中为我们提供... 查看详情

深度学习-计算机视觉-基础学习笔记-01(代码片段)

图像分类的思想:收集大量已知图像数据并带有正确的标签,将以往通过一个函数传入一张图片并返回图片的类型的这个思想转变成,通过俩个函数,第一个函数传入大量的图片并通过该函数训练返回一个模型,该... 查看详情

第62篇inception-v4

...3残差的缩放4训练方法5实验结果6结论摘要论文链接:Inception-v4近年来,非常深度的卷积网络一直是图像识别性能最大进步的核心。一个例子是Inception架构,它已经被证明以相对较低的计算成本实现了非常好的性能。... 查看详情

计算机视觉中图像分类任务脉络梳理

本文大致梳理了计算机视觉中图像分类的脉络,包括常用数据集、经典模型和性能对比。1图像分类常用数据集以下是几种常用的分类数据集,难度依次递增。列举了各算法在各数据集上的性能排名。MNIST,60k训练图像、10k测试图... 查看详情

tensorflow实战计算机视觉之猫狗大战

数据处理模型设计模型训练模型测试模型评估 查看详情

第62篇inception-v4

文章目录摘要1简介2相关工作3体系架构的选择3.1纯Inception块3.2剩余起始块3.3残差的缩放4训练方法5实验结果6结论 查看详情

ai:2021年人工智能前沿科技报告(更新中……)

...练模型成为下一个大模型重点发展领域。3.Transformer 成为计算机视觉领域的重要网络架构,并开始向强化学习、图神经网络等领域渗透。4.加速方法创新提升了超大参数规模模型的训练效率,催生更大规模参数的巨型模型... 查看详情

《计算机视觉基础知识蓝皮书》第9篇模型改进及期刊发表

本专栏将系统性地讲解计算机视觉基础知识、包含第1篇机器学习基础、第2篇深度学习基础、第3篇卷积神经网络、第4篇经典热门网络结构、第5篇目标检测基础、第6篇网络搭建及训练、第7篇模型优化方法及思路、第8篇模型超参... 查看详情

机器学习与tensorflow——tf.train.saver()inception-v3的应用(代码片段)

1.tf.train.Saver()tf.train.Saver()是一个类,提供了变量、模型(也称图Graph)的保存和恢复模型方法。TensorFlow是通过构造Graph的方式进行深度学习,任何操作(如卷积、池化等)都需要operator,保存和恢复操作也不例外。在tf.train.Saver()类初... 查看详情

开源bit:探索计算机视觉的大规模预训练

计算机视觉研究人员的普遍感觉是,现代深层神经网络总是渴望更多的标签数据-国家的最先进的电流细胞神经网络需要对接受培训数据集,例如OpenImages或Places,其中包含超过100万个标记图像。然而,对于许多应用程序,收集如... 查看详情

使用训练测试拆分找到最佳的超参数后,我应该用整个数据集重新训练模型吗?

我将数据集分为训练和测试。最后,在为训练数据集找到最佳的超参数之后,是否应该使用所有数据再次拟合模型?关键是要获得新数据的最高分数。答案是的,这将有助于推广模型,因为更多数据通常意味着更好的推广。 查看详情

『计算机视觉』normalization层(待续)

原文来自知乎一、两个概念独立同分布(independentandidenticallydistributed)独立同分布的数据可以简化常规机器学习模型的训练、提升机器学习模型的预测能力白化(whitening)去除特征之间的相关性—>独立;使得所有特征具有... 查看详情