点击下方卡片,关注 AIZOO 公众号
AI 重磅干货,第一时间送达
今天,分享一个新发布的基于 YOLOv5 的开源库,提供了多种 backbone和压缩工具,以及 FastAI 发布的一本电子书。
1. YOLOv5-Multibackbone-Compression 多 backbone以及压缩库
代码链接:
https://github/Gumpest/YOLOv5-Multibackbone-Compression
目前该仓库有 236 star,提供了多种不同的 backbone,包括以下backbone:
TPH-YOLOv5:加 Transformer 的 YOLOv5
SwinTrans-YOLOv5:SwinTrans 作为 backbone
YOLOv5l-GhostNet
YOLOv5l-ShuffleNetV2
YOLOv5l-MobileNetv3Small
YOLOv5l-EfficientNetLite0
YOLOv5l-PP-LCNet
各种 backbone 的 YOLOv5 在 Visdrone 数据集的精度如下:
(1 size is 608,2-6 size is 640)
No | Model | mAP | mAP@50 | Parameters(M) | GFLOPs | TrainCost(h) | Memory Cost(G) | PT File | FPS@CPU |
---|---|---|---|---|---|---|---|---|---|
1 | YOLOv5l | 23.2 | 39.7 | 46.19 | 108.1 | ||||
2 | YOLOv5l-GhostNet | 18.4 | 33.8 | 24.27 | 42.4 | 27.44 | 4.97 | PekingUni Cloud | |
3 | YOLOv5l-ShuffleNetV2 | 16.48 | 31.1 | 21.27 | 40.5 | 10.98 | 2.41 | PekingUni Cloud | |
4 | YOLOv5l-MobileNetv3Small | 16.55 | 31.2 | 20.38 | 38.4 | 10.19 | 5.30 | PekingUni Cloud | |
5 | YOLOv5l-EfficientNetLite0 | 19.12 | 35 | 23.01 | 43.9 | 13.94 | 2.04 | PekingUni Cloud | |
6 | YOLOv5l-PP-LCNet | 17.63 | 32.8 | 21.64 | 41.7 | 18.52 | 1.66 | PekingUni Cloud |
另外,该仓库还提供了各种减枝和量化、训练和评估工具,大大的利好学生啊。
2. FastAI 发布深度学习电子书
FastAI 是一个深度学习组织,发布了深度深度学习课程和开源代码,还发布了一个基于 PyTorch 的上层库 Fastai。书籍对应代码:https://github/fastai/fastbook
书籍下载地址:公众号 AIZOO 后台回复fastai
就可以下载。
书籍封面
书籍内容包括深度学习介绍、如何训练、深度学习在 CV和NLP的应用,以及如何部署,内容看着挺丰富的,大家可以自行下载和查看。
欢迎扫描下方的二维码添加小助手微信,邀请您加入我们的微信交流群。
群里有多位清北复交、BAT、AI独角兽大牛和众多深度学习er在一起愉快的交流技术,有任何问题,都可以咨询大家,欢迎你的加入哦。
添加小助手微信,邀您进 AIZOO 技术交流群
听说点个在看的人运气都很好~
更多推荐
今日AI:YOLOv5多种backbone、模型压缩代码库开源;FastAI 发布 PyTorch 学习免费书...
发布评论