上传者: 42166105
|
上传时间: 2021-10-19 21:17:16
|
文件大小: 17KB
|
文件类型: -
使用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