用tensorflow实现yolov3

2024-05-21 07:04:00

完整YOLOv3流程
原作者 : YunYang1994

论文： YOLOv3

环境： win7 + anaconda + python3.6 ( 建议用anaconda，方便一些)

下载代码之前，请先装好环境和相关依赖。

pip install -r docs/requirements.txt

1.github下载YOLOv3的tensorflow实现代码。

git clone git@github.com:sowei728/tensorflow_for_YOLOv3.git

2.下载YOLOv3的初始权值放到checkpoint文件夹中。

wget https://pjreddie.com/media/files/yolov3.weights
windows 没有wget命令就直接找个浏览器把URL输进去就好了

3.标注自己的VOC格式数据。

使用labelImg脚本，下载地址： http://tzutalin.github.io/labelImg/

脚本使用非常简单，自己看看就会了，最后脚本生成的xml文件放入dataset文件夹的anno文件夹中，原jpg的图片放入images文件夹中

4. 将xml文件中的标注信息汇总到txt中。

自己新建3个txt文件：

labels.txt 存放所有图片的标注信息

train.txt 存放训练图片的标注信息（总图片的80%）

test.txt 存放测试图片的标注信息（总图片的20%）
文件内容：(以空格隔开)

图片路径 x1min y1min x1max y1max cls1_id x2min y2min x2max y2max cls2_id …
运行脚本即可：

python scripts/load_xml_into_txt.py

5.将train.txt和test.txt转化为train.tfrecords和test.tfrecords文件。

python convert_tfrecord.py --dataset_txt ./apple_dataset/train.txt --tfrecord_path_prefix ./apple_dataset/apple_train

python convert_tfrecord.py --dataset_txt ./apple_dataset/test.txt --tfrecord_path_prefix ./apple_dataset/apple_test
apple_test.tfrecords 和 apple_train.tfrecords会生成在dataset目录下

6. 查看文件标注有无问题。

python show_input_image.py
如果正常出现图像及标注信息则代表文件标注无问题

7.kmeans求出k个anchor box的长宽。

python kmeans.py
会输出准确率和k个anchor box的长宽信息，本实验中确定的k值为9

长宽信息需要填入到data/apple_anchors.txt中

8.转化yolov3.weights为ckpt模型权值。

python convert_weight.py --convert

9.训练模型，保存模型到checkpoint文件夹中。

python quick_train.py

用tensorflow实现yolov3相关推荐

Tensorflow版yolov3训练自己的数据
Tensorflow版yolov3训练自己的数据源代码:https://github.com/YunYang1994/TensorFlow2.0-Examples/tree/master/4-Obj ...
YOLO学习-1：win10（64位）+ python3.6 + TensorFlow + keras + yolov3测试实践
实验环境:win10(64位).Python3.6(未安装anaconda版,用了pycharm).TensorFlow(未安装gpu版).keras.yolov3代码.yolov3权重文件第一步: ...
Win7+keras+tensorflow使用YOLO-v3训练自己的数据集
一.下载和测试模型 1. 下载YOLO-v3 git clone https://github.com/qqwweee/keras-yolo3.git 这是在Ubuntu里的命令,windows直接去 ...
tensorflow训练yolov3模型（检测雪人为例，自己的数据和标签，windows环境）
惯例先放效果所有代码包含训练.测试图片视频打包在: 地址下载代码: git clone https://github.com/YunYang1994/tensorflow-yolov3 或者点此下 ...
Yolo-v3 and Yolo-v2 for Windows and Linux 翻译
Yolo-v3 and Yolo-v2 for Windows and Linux(适用于Windows和Linux的Yolo-v3和Yolo-v2) (neural network for obje ...
YOLOv3 代码详解(2) —— 数据处理 dataset.py解析：输入图片增强、制作模型的每层输出的标签
前言: yolo系列的论文阅读论文阅读 || 深度学习之目标检测重磅出击YOLOv3 论文阅读 || 深度学习之目标检测yolov2 论文阅读 || 深度学习之目标检测yolov1 该篇讲解的 ...
keras版yolov3绘制acc和loss曲线
keras版yolov3绘制acc和loss曲线基本概念程序基本概念 loss表示损失函数值,随着训练过程的进行,loss会越来越小.acc表示模型预测准确率.写论文时,loss曲线和acc曲线 ...
tensorflow_yolo-v3笔记 IOU：Intersection over union 交并比
项目地址:YunYang1994/tensorflow-yolov3 Anchors clustering 关于IOU的解释:Intersection over union 参考文章1:Github ...
PaddlePaddle飞桨OCR文本检测——入门安装GPU版（一）
首先放上github上的开源代码网址,里面的帮助文档说的非常详细: https://github.com/PaddlePaddle/PaddleOCR/ 环境安装 python中环境使用pip安装: ...

最新文章

热门文章