tensorflow api训练2(准备数据集,训练)

2. 我们用的tensorflow mobile netv2官方的预训练模型

3. 准备预训练模型和数据集

首先要知道文件夹的目录格式
在这里插入图片描述
在这里插入图片描述
打开后看到的是下面的东西,要改的东西都已经标注了
在这里插入图片描述在这里插入图片描述
然后还要一个pbtxt
在这里插入图片描述
在这里插入图片描述

4. 接下来教你怎么改数据集,

我之前说的VOC2012里面的文件包含什么,Annotations是我们打的标签,JPEGImages是我们的图片,ImageSets里面的Main文件夹里面包含
在这里插入图片描述
就是我们对我们图片的名字打包放到这里

Annotations的xml文件要求的格式是
在这里插入图片描述
这里自己体会吧,folder必须是voc2012,path里面的图片位置必须要在VOC2012的JPEGImages,网上有很多该路径的方法,但是他们并没有先删除我们之前的path路径,而是直接怼上去(看下方),
在这里插入图片描述
我也没找到方法,然后我想到了一个最笨的方法,把图片和xml文件分别放到指定的文件夹,然后用labelImg去打开我们之前打好的标签,打开后,我们对之前的标签稍微移动一下,然后保存,这样path就会变成你想要的路径了,但是弄完这个步骤,会出现下方的问题
在这里插入图片描述
这个问题,好解决,我们只需要替代他就行了(我提醒一下,如果原来图片是1080p的,最好缩小一下,不然可能会报一些你不明白的错)
然后我们生产txt文件,我这边有脚本生成(我会将脚本打包,有需要的可以下载,也可私信我)

5. VOC2012文件夹的东西弄完之后,我们生成record

在这里插入图片描述

python object_detection/dataset_tools/create_pascal_tf_record.py --label_map_path=I:/persontrain/train/data/label_map.pbtxt --data_dir=I:/persontrain --year=VOC2012 --set=train --output_path=I:/persontrain/pascal_train.record

生成val的话,把train换成val就行了,然后把这两个文件放到data下面与pbtxt同目录,

6. 然后改config文件,把刚才我指出要改的路径加上去,record就是我们刚才生成的路径,至此,我们改的东西就差不多了

7. 开始训练东西(cmd的路径,都是research下面),我们用legacy下的train文件

在这里插入图片描述

8. 我们只需要传入3个3个数据

python object_detection/legacy/train.py --pipeline_config_path=I:\persontrain\train\models\ssd_mobilenet_v2_coco.config --train_dir=I:\persontrain\train\models\train --logtosderr

在这里插入图片描述
点回车,开始训练
在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
人工智能项目资料-基于Tensorflow,以MINST为数据训练手写数字识别模型,并调用打开摄像头的API,对拍摄到的图片中的一个或多个数字进行识别 【探索人工智能的宝藏之地】 无论您是计算机相关专业的在校学生、老师,还是企业界的探索者,这个项目都是为您量身打造的。无论您是初入此领域的小白,还是寻求更高层次进阶的资深人士,这里都有您需要的宝藏。不仅如此,它还可以作为毕设项目、课程设计、作业、甚至项目初期的立项演示。 【人工智能的深度探索】 人工智能——模拟人类智能的技术和理论,使其在计算机上展现出类似人类的思考、判断、决策、学习和交流能力。这不仅是一门技术,更是一种前沿的科学探索。 【实战项目与源码分享】 我们深入探讨了深度学习的基本原理、神经网络的应用、自然语言处理、语言模型、文本分类、信息检索等领域。更有深度学习、机器学习、自然语言处理和计算机视觉的实战项目源码,助您从理论走向实践,如果您已有一定基础,您可以基于这些源码进行修改和扩展,实现更多功能。 【期待与您同行】 我们真诚地邀请您下载并使用这些资源,与我们一起在人工智能的海洋中航行。同时,我们也期待与您的沟通交流,共同学习,共同进步。让我们在这个充满挑战和机遇的领域中共同探索未来!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

jun_军

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值