YOLOV3-TT100K: 基于YOLOV3训练的交通标志检测模型

非常感谢 bubbliiiing 分享的 yolo3-pytorch 的代码，里面包含了丰富的注释，非常便于理解.

---

The demo of yolov3-tt100k.

在原始代码基础上：

1. 修改数据加载类，支持CoCo格式（使用cocoapi）；
2. 修改数据增强；
3. validation增加mAP计算；
4. 修改anchor；

训练集

Tsinghua-Tencent 100K

下载的训练集主要包含train和test两部分，分别为6107和3073张图片。统计标注文件，共221类。详细统计每类标志个数，发现很多类的数量为0，所以清楚了部分数量为0的label，剩下类别为151，其中仍存在很多类数量<5.

训练集标签数量统计.

TT100k转为CoCo格式：

参考：

python scripts/tt100k2coco.py

测试

pretrained model 密码: 026r

下载到model_data,运行:python predict.py

结果

如上，mAP不高，分析原因，可能如下：

1. 数据集分辨率2048x2048，yolov3输入为608,且交通标志中存在很多小物体，原图resize到608，很多目标太小难以检测；
2. 某些类别数量过少；

可优化:

1. 借鉴YOLT方法检测小物体；
2. 数据集扩充/增强；
3. 使用更优秀的检测方法；
4. 改进loss，解决类别不均衡可参考