【文件属性】：
文件名称：YOLOv5-tf:使用TensorFlow 2实现YOLOv5
文件大小：17KB
文件格式：ZIP
更新时间：2021-04-06 15:32:24
tensorflow tf2 object-detection tensorflow2 yolov5
使用TensorFlow 2实现
火车
在config.py更改数据集路径和class_dict
在config.py选择版本
可选， python generate.py为您的数据集生成锚点并在config.py更改锚点
运行python train.py进行培训
测试
运行python test.py
数据集结构
├── Dataset folder
├── IMAGES
├── 1111.jpg
├── 2222.jpg
├── LABELS
├── 1111.xml
├── 2222.xml
├── train.txt
├── test.txt
笔记
xml文件应为PascalVOC格式
train.txt包含不带扩展名的图像名称
推荐（适用于docker用户）
docker pu
【文件预览】：
YOLOv5-tf-master
----nets()
--------nn.py(14KB)
----generate.py(6KB)
----LICENSE(11KB)
----utils()
--------config.py(2KB)
--------util.py(6KB)
--------dataset.py(4KB)
----README.md(1KB)
----test.py(2KB)
----train.py(3KB)