Yolov3训练自己的数据集(基于TensorFlow)

   日期:2020-09-02     浏览:118    评论:0    
核心提示:这里写自定义目录标题WARNING!准备数据创建文件名的索引合并图像文件与标注文件开始训练预测predict.pyWARNING!本教程所使用的的程序为下图所示版本(https://blog.csdn.net/weixin_44791964/article/details/103276106)基于Tensorflow -GPU2.0.0+Keras2.3.1准备数据准备好图片以及标注文件:分别存放在VOCdevkit/VOC2007/JPEGImages和Annotations文件夹。在mode

@TOC

WARNING!

本教程所使用的的程序为下图所示版本(https://blog.csdn.net/weixin_44791964/article/details/103276106)

基于Tensorflow -GPU2.0.0+Keras2.3.1

准备数据

准备好图片以及标注文件:分别存放在VOCdevkit/VOC2007/JPEGImages和Annotations文件夹。
在model_data文件夹下面创建一个自己的分类txt文件(我这里以我创建的new-classes.txt为例)这个记事本里面存放的是数据集分类的类别名称(最后一行不能有空格!!!
将预训练的权重文件保存在model_data文件夹下面。

创建文件名的索引

在VOCdevkit/VOC2007文件夹中的voc2yolo3文件运行即可(记得在每次运行前手动删除VOCdevkit/VOC2007/Imagesets/Main里面的四个txt文件以及根目录下的2007_train test val三个文件)

合并图像文件与标注文件

合并图像文件与标注文件。修改
为自己的数据集的类别名称,然后运行该程序。若出现一下错误:
注意检查自己的标签和路径中不要使用中文。(这边要注意下:如果自己将这些文件都生成好了cp到服务器上面运行的时候一定会报错,就是因为这个2007_train.txt里面的路径是自己电脑上的索引,所以必须在服务器上面生成这三个文件,这样在训练的时候,程序才能找到图片以及对应的标注文件。)

开始训练

主要在程序里面修改batchsize以及epoch。这个程序冻结部分网络结构,以防止初期权值被破坏,并且可以解冻后再训练。
ATTENTION:我这里训练采用的是yolov3作者给的预训练好的权重文件。

预测predict.py

在训练完成之后,/logs文件夹下面会生成几个.h5的权重文件。其中last1.h5是最终的权重文件,需要我们cp到model_data文件夹里面。修改yolo.py文件夹中的路径文件名

model_path为训练好的权重文件的路径

 
打赏
 本文转载自:网络 
所有权利归属于原作者,如文章来源标示错误或侵犯了您的权利请联系微信13520258486
更多>最近资讯中心
更多>最新资讯中心
0相关评论

推荐图文
推荐资讯中心
点击排行
最新信息
新手指南
采购商服务
供应商服务
交易安全
关注我们
手机网站:
新浪微博:
微信关注:

13520258486

周一至周五 9:00-18:00
(其他时间联系在线客服)

24小时在线客服