文件名称:YOLOv5-tf:使用TensorFlow 2实现YOLOv5
文件大小:17KB
文件格式:ZIP
更新时间:2024-04-26 09:19:04
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)