Day20 - yolov5s-tph在VisDrone下的表现
昨天的acmix一直调不通代码，总是报RuntimeError: Input type (float) and bias type (c10::Half) should be the same，查了一下是因为model在验证的时候half了，参考https://github.com/LeapLabTHU/ACmix/issues/18#issuecomment-1730733557
决定换成transformer head，把yolov5里P345的C3模块换成C3TR，结果如下，以yolov5s作为baseline作对比
VisDrone，200epochs
| Model | Parameters(M) | mAP@50:95(%) | mAP50(%) |
|---|---|---|---|
| yolov5s | 9.15 | 21.12 | 36.06 |
| yolov5s-tph | 9.15 | 22.22 | 37.77 |
可以发现几乎没有变化，map降低了一个点，很无语，有点玄学感觉
本文由作者按照 CC BY 4.0 进行授权