fanhuafeng / yolov5_in_tf2_keras

tf2-keras implement yolov5

YOLOv5 in tesnorflow2.x-keras

模型测试

• 训练 COCO2017(val 5k)

• 检测效果

• 精度/召回率

Requirements

pip3 install -r requirements.txt

Get start

1. 训练

python3 train.py

2. tensorboard

tensorboard --host 0.0.0.0 --logdir ./logs/ --port 8053 --samples_per_plugin=images=40

3. 查看

http://127.0.0.1:8053

训练自己的数据

1. labelme打标自己的数据, 导出为coco格式

2. 参考CoCoDataGenrator类实现自己的generator

3. python3 yolov3.py 训练你的数据