本文将PicoDet模型网络结构迁移到yolov5的平台,因为是anchor base的形式,在性能上与原生模型可能有一定的差异,以下是原生模型的性能指标。...一、PicoDet介绍
Picodet论文在11月放出,后面对模型结构进行了复现,但因为没有显卡,迟迟无法对复现后的模型进行效果检验(现在依旧没有显卡可以跑模型,遂放出代码),这个周末有空,把之前的代码翻出来整理一下..., [ 464, 5, 2 ] ], # 26 (P5/32-large)
[ [ 17, 20, 23, 25 ], 1, Detect, [ nc, anchors ] ], # Detect...(P3, P4, P5, P6)
]
总的模型结构以及640x640下的param和Flops:
代码将更新于个人repo,欢迎大家star和白嫖!...https://github.com/ppogg/YOLOv5-Lite